Commit Graph
160 Commits
Author SHA1 Message Date
Matt Tancik 076c5bf0b7 Add stratified sampling and enable pdf sampling 2022-04-29 15:37:33 -07: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 47691eb291 removing goat references 2022-04-29 16:36:47 +00:00
Ethan Weber 91c6c2d899 Merge pull request #43 from ethanweber/ethan/apr29_using_real_test_set
Validation images integration.
2022-04-29 09:27:39 -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
evonneng 58f2622f4d added general testing, fixed up config, made writer general 2022-04-28 23:21:01 +00:00
evonneng e5f436a18d update description in yaml 2022-04-28 22:09:00 +00:00
evonneng 609fcb04f6 Set up profiler and moved stats tracker to its own class; set up configurables for stats tracker and profiler (#42)
* 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
2022-04-28 15:04:00 -07:00
Matthew Tancik 2e350c148f Merge pull request #41 from ethanweber/tancik/fix_encoding
Append input to encoding
2022-04-27 12:12:25 -07:00
Matt Tancik 69fa901788 Append input to encoding 2022-04-27 12:08:39 -07:00
evonneng 23b70634cb Remove clutter for logging in terminal (#40)
* 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
2022-04-26 12:12:57 -07:00
evonneng 5d4ffa210a Prettier printing of times. (#39)
* 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
2022-04-25 09:17:19 -07:00
Ethan Weber 9912c812c0 Fixing Tuple converting to int error. 2022-04-25 08:45:35 -07:00
Ethan Weber 11d9c7271a black 2022-04-23 19:51:05 +00:00
Ethan Weber daf5a6251a Merge pull request #38 from ethanweber/ethan/caching_for_dataloader
Added caching to the dataloader for faster training.
2022-04-23 12:48:28 -07:00
Ethan Weber 80ad17700c pylint and test cases for dataloader 2022-04-23 19:45:53 +00:00
Ethan Weber 391803759e adding config 2022-04-23 17:37:13 +00:00
Ethan Weber 4c7d07116e added caching for dataloader 2022-04-23 17:36:57 +00:00
Ethan Weber 5f1baa5a7b Merge branch 'master' of github.com:ethanweber/mattport 2022-04-23 15:26:57 +00:00
Ethan Weber 078dc6aa04 removing dotdict 2022-04-23 15:26:53 +00:00
evonneng ca76f70d49 Added timing logic within trainer class (#37)
* 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
2022-04-22 17:33:02 -07:00
Ethan Weber 0dcf2ae599 fixed writer 2022-04-23 00:12:34 +00:00
Matthew Tancik 91fea71937 Merge pull request #36 from ethanweber/tancik/fix_camera_index_bug
Fix camera index bug
2022-04-22 16:57:33 -07:00
Matt Tancik 391fd93247 Fix camera index bug 2022-04-22 16:56:50 -07:00
Ethan Weber 181438a88b Merge branch 'ethan/test_set_integration' 2022-04-22 18:19:57 +00:00
Ethan Weber 5099a4f7c5 resizing image 2022-04-22 18:19:39 +00:00
Ethan Weber a5b09b0769 Merge pull request #35 from ethanweber/ethan/test_set_integration
Fake version of test set with tensorboard.
2022-04-21 20:56:38 -07:00
Ethan Weber f090d9a001 training loop with tensorboard funs 2022-04-22 03:52:35 +00:00
Ethan Weber 4c0df2fde8 merged from master 2022-04-22 03:51:28 +00:00
Ethan Weber 9cc9f0f233 Merge branch 'master' of github.com:ethanweber/mattport 2022-04-22 03:45:19 +00:00
Ethan Weber cf7034f0d1 added test set (using training data for now) 2022-04-22 03:45:06 +00:00
Ethan Weber 7e5a5bf524 code runs again, working on integrating test set and optimizers 2022-04-21 21:57:36 +00:00
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 8ed28b7a37 Merge branch 'master' of github.com:ethanweber/mattport 2022-04-21 20:29:45 +00:00
Ethan Weber b6f878f227 merge 2022-04-21 20:29:43 +00: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