Ruilong Li(李瑞龙)
12cd321e1a
Update Instant-NGP & Density Grid with CUDA supports. Much faster and better PSNR. ( #193 )
2022-07-24 22:22:44 -07:00
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
Matthew Tancik
74280110ce
Remove redundant data ( #103 )
...
RaySamples contained fields for bin_starts and bin_ends. These are also stored in the frustums. This PR removes them from the RaySamples.
2022-07-07 18:07:34 -07:00
Matt Tancik
bd78a68756
Update RaySamples to TensorDataclass
2022-07-06 12:01:09 -07:00
Matthew Tancik
59957fa202
Use tensordataclasses ( #85 )
...
+ Convert Frustum and RayBundle into TensorDataclasses.
+ Support TensorDataclass fields in TensorDataclasses
+ Support non batched types in TensorDataclass (The are iignored)
2022-07-05 18:32:43 -07:00
Ethan Weber
fefbad2592
Fixed frustum device and updated model checkpoint defaults ( #78 )
...
* fix tensor device and update default model save dir
* removed viewer reference
2022-07-04 17:22:39 -07:00
Matt Tancik
5a1c8a1c86
Remove point samples
2022-07-01 09:36:59 -07:00
evonneng
91ef54963c
refactoring configs; pulled network out of graph; created trainer heirarchy ( #51 )
...
* changed config graph structure
* removed network udner graphs; added new heirarchy trainer
* fixed callback in occupancy graph
* lint
* changed register callback to setter method
2022-06-15 09:06:43 -07:00
Matt Tancik
37a3692b45
Update docstrings
2022-06-10 09:03:43 -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