messed up by removing not

stefanklut · Apr 9, 2024 · 9d0aa83 · 9d0aa83
1 parent 195badf
commit 9d0aa83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -108,7 +108,7 @@ def setup_training(args: argparse.Namespace):
 
         trainer = Trainer(cfg=cfg)
         if not cfg.TRAIN.WEIGHTS:
-            if cfg.MODEL.RESUME and trainer.checkpointer.has_checkpoint():
+            if cfg.MODEL.RESUME and not trainer.checkpointer.has_checkpoint():
                 raise FileNotFoundError(f"No checkpoint found in {cfg.OUTPUT_DIR}")
             trainer.resume_or_load(resume=cfg.MODEL.RESUME)
         else: