Commit Graph
11 Commits
Author SHA1 Message Date
Ethan Weber b6f878f227 merge 2022-04-21 20:29:43 +00:00
evonneng 4d1a6b2dbd cleanup and added logic to train for load/save checkpoints (#32)
* added checkpointing logic

* small nits

* check DDP running correctly

* cleanup and added logic to train for load/save checkpoints

* lint
2022-04-21 11:39:44 -07:00
Ethan Weber be6fbe799e most code for nerf running 2022-04-19 22:32:10 +00:00
evonneng b8e2713643 Changed Graph to nn.ModuleDict; Added DDP logic to seup_graph; locallyrunning DDP 2022-04-15 23:35:00 +00:00
evonneng 62d4218421 updated logic within Trainer class to allow for multiprocess training 2022-04-15 21:53:35 +00:00
evonneng ac3e7515ef done adding main logic in run_train_nerf to initialize processes with more robust handling of DDP 2022-04-15 21:33:50 +00:00
evonneng 7dc9d445b2 added basic distributed logic for trainer 2022-04-15 01:04:17 +00:00
Ethan Weber 5691d00d42 fixed most linting errors 2022-04-07 20:22:48 +00:00
Ethan Weber a7167f890e brainstorming trainer code but broken 2022-04-06 22:45:53 +00:00
Ethan Weber a9c0d5b97f hydra config and starting trainer class 2022-04-06 16:46:47 +00:00
Ethan Weber b63d5854a1 reworked structure to use packages 2022-04-02 20:07:41 +00:00