From 9d0aa8363f3dc7bbc5b470627d833bec8ed5c3ad Mon Sep 17 00:00:00 2001 From: Stefan Klut Date: Tue, 9 Apr 2024 16:54:15 +0200 Subject: [PATCH] messed up by removing not --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index b6953d1..1eaa706 100644 --- 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: