fix viewer checkpointing (#511)

This commit is contained in:
Ethan Weber authored and GitHub committed 2022-09-21 17:56:18 -07:00
1 parent a1152be22b
commit 21ebaceed1
4 files changed
+12 -7

No files matched your search

+1 -1
View File
@@ -102,7 +102,7 @@ def _eval_setup(config_path: Path) -> Tuple[cfg.Config, Pipeline, Path]:
# load checkpoints from wherever they were saved
# TODO: expose the ability to choose an arbitrary checkpoint
config.trainer.load_dir = config.trainer.model_dir
config.trainer.load_dir = config.get_checkpoint_dir()
config.pipeline.datamanager.eval_image_indices = None
# setup pipeline (which includes the DataManager)