Ethan Weber
023e6a3f92
merging masters new config
2022-04-29 22:29:43 +00:00
Ethan Weber
387ad93027
merging
2022-04-29 22:10:36 +00:00
evonneng
a93dace958
modularizing hydra config and setting defaults ( #46 )
...
* lint
* reconfigured modules in yaml
2022-04-29 14:39:43 -07:00
Ethan Weber
7b118750ec
merged master into this
2022-04-29 21:37:10 +00:00
Ethan Weber
6b17ad51c6
cpu support and multi gpu progress
2022-04-29 21:29:13 +00:00
evonneng
86b1011ae1
Printing profile on sig kill. ( #44 )
...
* added skeleton for terminal logger
* done skeleton terminal writer logic... todo: actual calls to the write classes
* moved lambda to top function
* added logic to support timing
* prettier printing
* prettier printing
* lint
* separated out timing and other stats variables. printing in dict format
* keeping logging history and wiping rest
* updated logic to calculate total times instead of average times
* made stats tracker configurable
* moved stats tracker and profiler to own class file; implemented basic functionality for profiler
* lint removing duplicate function
* update readme; moved around config variables
* profiler prints on sig kill
2022-04-29 11:10:34 -07:00
Ethan Weber
dd6f79fc40
using test images. including some local data. black on save. disabled warnings for pytest
2022-04-29 16:23:05 +00:00
Ethan Weber
cf7034f0d1
added test set (using training data for now)
2022-04-22 03:45:06 +00:00
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
0b7dfcdc5d
Initial ray generation and dataset code ( #12 )
...
* ray generator and cameras structure progress
* test renames
* added pass
* conflicts
* fixing some black errors
* reformated files
* merged master and .gitignore
* fixing linting
* removing old folder
* removed return False
* pylint fixes
* update
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
2022-04-12 15:08:16 -07: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
Ethan Weber
6aae70df32
first commit
2022-03-29 17:26:54 +00:00
