Add default hyperparams

dshabin · dshabin · commit 820bae7c860d · 2021-07-14T15:19:18.000+04:00
diff --git a/superannotate/ml/defaults.py b/superannotate/ml/defaults.py
@@ -1,7 +1,6 @@
 DEFAULT_HYPERPARAMETERS = {
     "instance_type": "1 x T4 16 GB",
     "num_epochs": 12,
-    "dataset_split_ratio": 80,
     "base_lr": 0.02,
     "gamma": 0.5,
     "images_per_batch": 4,
diff --git a/superannotate/ml/ml_funcs.py b/superannotate/ml/ml_funcs.py
@@ -301,6 +301,8 @@ def run_training(
             0,
             f"The type of provided projects is {project_type}, and does not correspond to the type of provided model"
         )
+    if not hyperparameters:
+        hyperparameters = {}
     for item in DEFAULT_HYPERPARAMETERS:
         if item not in hyperparameters:
             hyperparameters[item] = DEFAULT_HYPERPARAMETERS[item]

Original file line number	Diff line number	Diff line change
`@@ -301,6 +301,8 @@ def run_training(`
`301`	`301`	`0,`
`302`	`302`	`f"The type of provided projects is {project_type}, and does not correspond to the type of provided model"`
`303`	`303`	`)`
	`304`	`+ if not hyperparameters:`
	`305`	`+ hyperparameters = {}`
`304`	`306`	`for item in DEFAULT_HYPERPARAMETERS:`
`305`	`307`	`if item not in hyperparameters:`
`306`	`308`	`hyperparameters[item] = DEFAULT_HYPERPARAMETERS[item]`