Update config.json

supervisely-ecosystem · Jul 17, 2023 · 1d60c89 · 1d60c89
1 parent 9eb2584
commit 1d60c89
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/supervisely/train/config.json b/supervisely/train/config.json
@@ -15,7 +15,8 @@
   "main_script": "supervisely/train/src/sly_train.py",
   "gui_template": "supervisely/train/src/gui.html",
   "task_location": "workspace_tasks",
-  "gpu": "required",
+  "need_gpu": true,
+  "gpu": "preferred",
   "icon": "https://i.imgur.com/HZCM0QM.png",
   "icon_cover": true,
   "context_menu": {
@@ -30,4 +31,4 @@
   "license": {
     "type": "AGPL-3.0"
   }
-}
+}