most code for nerf running

This commit is contained in:
Ethan Weber committed 2022-04-19 22:32:10 +00:00
1 parent c2f3c06b6c
commit be6fbe799e
16 files changed
+223 -168

No files matched your search

+1 -1
View File
@@ -112,7 +112,7 @@ def _train(local_rank: int, world_size: int, config: DictConfig) -> Any:
Returns:
Any: TODO(): determine the return type
"""
trainer = Trainer(local_rank, world_size, config)
trainer = Trainer(config, local_rank, world_size)
trainer.setup_graph()
trainer.setup_optimizer()
trainer.setup_dataset()