Setting near and far with scene bounds collider. Also LR schedules. (#74)

* some initial cuda code as I learn. using git to save progress.

* first cuda kernel for ray sampling that mostly works

* cleaning up code. working on lr. structs for scene bounds aabb

* added scene bounds collider code

* updated setup

* setup

* setup

* black

* including some files

* exporting env variable

* new file

* git workflows

* github actions

* github actions

* github actions

* removing env var
This commit is contained in:
Ethan Weber authored and GitHub committed 2022-05-16 14:03:19 -07:00
1 parent 752df417ce
commit 676beebb14
45 files changed
+803 -8354

No files matched your search

+1 -1
View File
@@ -56,7 +56,7 @@ def main():
trainer = Trainer(config, local_rank=0, world_size=1)
trainer.setup(test_mode=True)
avg_psnr = 0
for step, image_idx in enumerate(config.data.validation_image_indices):
for step, image_idx in enumerate(config.data.val_image_indices):
with torch.no_grad():
psnr = trainer.test_image(image_idx=image_idx, step=step)
avg_psnr = (step * avg_psnr + psnr) / (step + 1)