Update SDK and Docker version

supervisely-ecosystem · Jan 16, 2025 · 6ae3629 · 6ae3629
1 parent 5c6933d
commit 6ae3629
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 6 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -38,4 +38,4 @@ RUN mkdir -p ~/.cache/supervisely/checkpoints && \
     curl -L -o ~/.cache/supervisely/checkpoints/rtdetrv2_r50vd_6x_coco_ema.pth https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetrv2_r50vd_6x_coco_ema.pth && \
     curl -L -o ~/.cache/supervisely/checkpoints/rtdetrv2_r101vd_6x_coco_from_paddle.pth https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetrv2_r101vd_6x_coco_from_paddle.pth
 
-RUN pip install --no-cache-dir supervisely[training]==6.73.279
+RUN pip install --no-cache-dir supervisely[training]==6.73.280
diff --git a/docker/publish.sh b/docker/publish.sh
@@ -1,2 +1,2 @@
-docker build -t supervisely/rt-detrv2:1.0.6 . && \
-docker push supervisely/rt-detrv2:1.0.6
+docker build -t supervisely/rt-detrv2:1.0.7 . && \
+docker push supervisely/rt-detrv2:1.0.7
diff --git a/supervisely_integration/serve/config.json b/supervisely_integration/serve/config.json
@@ -13,7 +13,7 @@
 	"gpu": "preferred",
 	"session_tags": ["deployed_nn"],
 	"community_agent": false,
-	"docker_image": "supervisely/rt-detrv2:1.0.6",
+	"docker_image": "supervisely/rt-detrv2:1.0.7",
 	"task_location": "application_sessions",
 	"license": {
 		"type": "Apache-2.0"

diff --git a/supervisely_integration/train/config.json b/supervisely_integration/train/config.json
@@ -9,7 +9,7 @@
 	"icon_cover": true,
 	"poster": "https://github.com/user-attachments/assets/b4554de5-3d2c-4b4f-aba9-95864cb05289",
 	"description": "Train RT-DETRv2 model on your data",
-	"docker_image": "supervisely/rt-detrv2:1.0.6",
+	"docker_image": "supervisely/rt-detrv2:1.0.7",
 	"instance_version": "6.12.17",
 	"task_location": "workspace_tasks",
 	"need_gpu": true,

diff --git a/supervisely_integration/train/main.py b/supervisely_integration/train/main.py
@@ -48,7 +48,6 @@ def start_training():
     solver.fit()
     # gather experiment info
     experiment_info = {
-        "task_type": sly.nn.TaskType.OBJECT_DETECTION,
         "model_name": train.model_name,
         "model_files": {"config": model_config_path},
         "checkpoints": output_dir,