mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 02:45:34 +08:00
most code for nerf running
This commit is contained in:
1 parent
c2f3c06b6c
commit
be6fbe799e
16 files changed
+223
-168
No files matched your search
@@ -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()
|
||||
|
||||
Reference in new issue
Block a user