fixing eval dataloader indices issue (#196)

* added some logic for the indices when none

* forgot to comment out code

* fix linting

* fixed typing
This commit is contained in:
Ethan Weber authored and GitHub committed 2022-07-22 16:29:14 -07:00
1 parent 76c8c178be
commit 6cfccab336
3 files changed
+13 -3

No files matched your search

+1
View File
@@ -41,6 +41,7 @@ def _load_best_ckpt(hydra_dir: str, config: DictConfig) -> str:
step = os.path.splitext(latest_ckpt)[0].split("-")[-1]
config.resume_train.load_dir = model_dir
config.resume_train.load_step = int(step)
config.data.dataloader_eval.image_indices = None
return os.path.join(model_dir, latest_ckpt)