We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b9af090 commit 4dcac5fCopy full SHA for 4dcac5f
imgx/imgx/run_train.py
@@ -156,7 +156,7 @@ def main( # pylint:disable=too-many-statements
156
):
157
ckpt_dir = files_dir / "ckpt" / f"batch_{i}"
158
159
- if config.eval and (ckpt_dir is not None):
+ if ckpt_dir is not None:
160
# when diffusion save under sampler name
161
out_dir = ckpt_dir
162
if config.task.name == "diffusion_segmentation":
0 commit comments