Matthew Tancik
441dcd5314
Add mix precision training ( #172 )
...
* Add mix precision training
* Disable mixed precision for cpu
2022-07-20 13:21:06 -07:00
Matt Tancik
f77d1fc110
Update docstrings and add missing types
2022-07-19 16:12:05 -07:00
evonneng
f67dca6f5b
refactor moving setup functions to associated classes; cleanup eval and benchmark scripts ( #48 )
...
* refactor code to move graph/dataset setup to respective files
* consolidated eval and benchmark script
* moved setup optimizers to its own thing
2022-06-13 19:05:07 -07:00
evonneng
5433f831ac
add licensing headers ( #27 )
...
* add licensing headers
* add licensing to run actions
* rerun headers file
* remove trailing white spaces
2022-06-08 13:31:13 -07:00
Ethan Weber
457788d689
dataloader logic, new folder structure, flatter configs ( #26 )
...
* renaming dataloader to data. adding some dataloader classes
* restructuring configs
* fixed instant ngp tcnn direction rescaling
* working through test cases
* black and lint and tests
2022-06-08 09:01:56 -07:00
Matthew Tancik
evonneng
Ethan Weber