Skip to content

Commit 4dcac5f

Browse files
authored
Update run_train.py
Remove non-existing `config.eval`
1 parent b9af090 commit 4dcac5f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

imgx/imgx/run_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def main( # pylint:disable=too-many-statements
156156
):
157157
ckpt_dir = files_dir / "ckpt" / f"batch_{i}"
158158

159-
if config.eval and (ckpt_dir is not None):
159+
if ckpt_dir is not None:
160160
# when diffusion save under sampler name
161161
out_dir = ckpt_dir
162162
if config.task.name == "diffusion_segmentation":

0 commit comments

Comments
 (0)