fix weights loading cfg

PatBall1 · Jul 21, 2022 · 5f3aabc · 5f3aabc
1 parent 1c2d971
commit 5f3aabc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detectree2/models/train.py b/detectree2/models/train.py
@@ -328,7 +328,7 @@ def setup_cfg(
   cfg.OUTPUT_DIR = out_dir
   os.makedirs(cfg.OUTPUT_DIR, exist_ok=True)
   if update_model is not None:
-    cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(update_model)
+    cfg.MODEL.WEIGHTS = update_model # DOESN'T WORK
   else:
     cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(base_model)