Commit Graph
126 Commits
Author SHA1 Message Date
Matthew Tancik f75d7b28cf Merge pull request #34 from ethanweber/tancik/nerf_restructure
Rearrange nerf files
2022-04-21 14:54:25 -07:00
Matt Tancik a32c681a7a Rearrange nerf files 2022-04-21 14:51:51 -07:00
Ethan Weber 0c3a72b8b3 Merge pull request #31 from ethanweber/tancik/dict_outputs
Switch to dictionary outputs
2022-04-21 13:28:50 -07:00
evonneng 17d837babf Integrating logic for setting up optimizers in run_train and config files (#33)
* added optimizers

* added optimizer logic for setting it up; todo-- fix optimizers and add step logic

* typing out the alphabet
2022-04-21 13:23:18 -07:00
Matt Tancik 326581d1a8 Merge branch 'tancik/dict_outputs' of github.com:ethanweber/mattport into tancik/dict_outputs 2022-04-21 11:43:31 -07:00
Matt Tancik 755716dbf7 Merge branch 'master' of github.com:ethanweber/mattport into tancik/dict_outputs 2022-04-21 11:42:39 -07:00
Matt Tancik 631a694f6b fix lint errors 2022-04-21 11:41:15 -07:00
Matt Tancik 05f2537605 Switch to dictionary outputs 2022-04-21 11:40:05 -07: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
Matt Tancik 7f32bb91ec Switch to dictionary outputs 2022-04-21 11:31:33 -07:00
evonneng af74b4e95c check DDP running correctly 2022-04-21 17:06:29 +00:00
evonneng 1cb3a02f07 added optimizers (#30) 2022-04-20 17:51:13 -07:00
Matthew Tancik 954414ee42 Merge pull request #29 from ethanweber/tancik/file_pipeline
Tancik/rough_pipeline
2022-04-20 14:14:48 -07:00
Matt Tancik 32af0082c7 Fix linting issues 2022-04-20 14:10:59 -07:00
Matt Tancik 10761460de Update lint config 2022-04-20 14:05:45 -07:00
Matt Tancik 3311b3c94e run black 2022-04-20 14:02:44 -07:00
Matt Tancik 40f0338eb2 Get tests passing 2022-04-20 13:58:39 -07:00
evonneng 2626f6acd9 Eng/logger (#28)
* added skeleton for terminal logger

* done skeleton terminal writer logic... todo: actual calls to the write classes

* moved lambda to top function
2022-04-20 13:43:36 -07:00
Matt Tancik 4325c86997 Get code running 2022-04-20 12:24:08 -07:00
Matt Tancik 5ee9669b2c Add sampler 2022-04-20 12:24:08 -07:00
evonneng bf3b77f952 Eng/distributed training (#27)
* added checkpointing logic

* small nits
2022-04-20 12:06:29 -07:00
Ethan Weber ba7a2c33c4 Merge pull request #26 from ethanweber/eng/logger
Eng/logger
2022-04-19 21:53:45 -07:00
evonneng 5b629b9495 added writer with declarator to checkif you are on main thread; add write image 2022-04-20 04:45:49 +00:00
Ethan Weber 862d8047b3 Merge pull request #25 from ethanweber/vanilla_nerf_implementation
A (not-quite-correct) version of NeRF strung together.
2022-04-19 20:28:42 -07:00
Ethan Weber 7ff63fe0be Merge branch 'master' into vanilla_nerf_implementation 2022-04-20 03:27:52 +00:00
Ethan Weber 0b0f7edb62 data instructions 2022-04-20 03:23:35 +00:00
Ethan Weber 9cd5c58fbd black formatting 2022-04-20 03:18:27 +00:00
Ethan Weber 7819f6da9d network pass running 2022-04-20 03:17:19 +00:00
Ethan Weber be6fbe799e most code for nerf running 2022-04-19 22:32:10 +00:00
evonneng 89a7d1ba53 added base writer class and tensorboard support; todo: init from config in trainer.py 2022-04-19 07:02:31 +00:00
Matthew Tancik dbdfd01841 Merge pull request #24 from ethanweber/tancik/mlp_test
Updated mlp tests
2022-04-18 17:56:35 -07:00
Matt Tancik aa2525e5cb Update imports 2022-04-18 17:54:29 -07:00
Matt Tancik 83c664e65a Updated mlp tests 2022-04-18 17:51:39 -07:00
Ethan Weber c2f3c06b6c brainstorming with matt on vanilla nerf implementation 2022-04-19 00:45:30 +00:00
Ethan Weber 0ac2708959 Merge pull request #22 from ethanweber/eng/distributed_training
Eng/distributed training
2022-04-17 09:48:00 -07:00
Ethan Weber c85ee76b1f removed build_dataset 2022-04-17 16:45:38 +00:00
Ethan Weber e5bc28c4fb Merge branch 'master' into eng/distributed_training 2022-04-17 09:40:30 -07:00
Ethan Weber 621a94e65c removing docker and fixing nfs issue 2022-04-17 16:36:09 +00:00
Ethan Weber e38040ee4d Image dataset implemented and ray generation working. (#23)
* working data loader code

* adding gitignore

* fixing linting errors

* set black line length in settings file

* update

* fixed filepath

* black format

* tqdm
2022-04-17 09:26:44 -07:00
evonneng 563dba0bad lint 2022-04-16 01:25:55 +00:00
Matthew Tancik 9bc83b0ed2 Merge pull request #21 from ethanweber/tancik/type_check
Add type checking
2022-04-15 17:57:19 -07:00
Matt Tancik 5da5a0deac Add type checking 2022-04-15 17:53:22 -07:00
Matthew Tancik 5df1748666 Merge pull request #20 from ethanweber/tancik/renderers_test
Fix renderer, add test
2022-04-15 17:04:17 -07:00
Matt Tancik a96088054a Fix renderer, add test 2022-04-15 16:58:28 -07: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 b2220fef28 Merge branch 'master' of github.com:ethanweber/mattport into eng/distributed_training 2022-04-15 22:22:07 +00:00
evonneng 8e36b8c313 moved tests to their own directory (#19) 2022-04-15 15:19:04 -07:00
Matthew Tancik eea1b61657 Merge pull request #18 from ethanweber/tancik/losses
Add basic rendering losses
2022-04-15 15:00:23 -07:00
Matt Tancik 1ee3cab281 Add basic rendering losses 2022-04-15 14:58:23 -07:00
evonneng 62d4218421 updated logic within Trainer class to allow for multiprocess training 2022-04-15 21:53:35 +00:00