Update run_train.py

Remove non-existing `config.eval`
mathpluscode · Oct 1, 2023 · 4dcac5f · 4dcac5f
1 parent b9af090
commit 4dcac5f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/imgx/imgx/run_train.py b/imgx/imgx/run_train.py
@@ -156,7 +156,7 @@ def main(  # pylint:disable=too-many-statements
         ):
             ckpt_dir = files_dir / "ckpt" / f"batch_{i}"
 
-        if config.eval and (ckpt_dir is not None):
+        if ckpt_dir is not None:
             # when diffusion save under sampler name
             out_dir = ckpt_dir
             if config.task.name == "diffusion_segmentation":