diff --git a/supervisely/train/src/sly_init_ui.py b/supervisely/train/src/sly_init_ui.py index 39a7b233e64f..a36940f5a4c2 100644 --- a/supervisely/train/src/sly_init_ui.py +++ b/supervisely/train/src/sly_init_ui.py @@ -83,8 +83,8 @@ def init_model_settings(data, state): state["pretrainedWeights"] = f'{data["modelSizes"][0]["label"]}.pt' # @TODO: for debug - state["weightsPath"] = "/yolov5_train/coco128_002/2390/weights/best.pt" - #state["weightsPath"] = "" + #state["weightsPath"] = "/yolov5_train/coco128_002/2390/weights/best.pt" + state["weightsPath"] = "" def init_training_hyperparameters(state):