423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-14 19:36:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

lazy all_train_cameras

Browse Source 
Summary: Avoid calculating all_train_cameras before it is needed, because it is slow in some datasets.

Reviewed By: shapovalov

Differential Revision: D38037157

fbshipit-source-id: 95461226655cde2626b680661951ab17ebb0ec75
This commit is contained in:
Jeremy Reizenstein
2022-07-21 15:04:00 -07:00
committed by Facebook GitHub Bot
parent b2dc520210
commit 3783437d2f
6 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
projects/implicitron_trainer/experiment.py
View File

@@ -391,7 +391,6 @@ def run_training(cfg: DictConfig) -> None:
datasource = ImplicitronDataSource(**cfg.data_source_args)
datasets, dataloaders = datasource.get_datasets_and_dataloaders()
task = datasource.get_task()
all_train_cameras = datasource.get_all_train_cameras()
# init the model
model, stats, optimizer_state = init_model(cfg=cfg, accelerator=accelerator)
@@ -405,7 +404,7 @@ def run_training(cfg: DictConfig) -> None:
_eval_and_dump(
cfg,
task,
all_train_cameras,
datasource.all_train_cameras,
datasets,
dataloaders,
model,
@@ -490,7 +489,7 @@ def run_training(cfg: DictConfig) -> None:
):
_run_eval(
model,
all_train_cameras,
datasource.all_train_cameras,
dataloaders.test,
task,
camera_difficulty_bin_breaks=cfg.camera_difficulty_bin_breaks,
@@ -525,7 +524,7 @@ def run_training(cfg: DictConfig) -> None:
_eval_and_dump(
cfg,
task,
all_train_cameras,
datasource.all_train_cameras,
datasets,
dataloaders,
model,