mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-14 03:44:02 +08:00
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:
1 parent
76c8c178be
commit
6cfccab336
3 files changed
+13
-3
No files matched your search
@@ -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)
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user