Commit Graph
31 Commits
Author SHA1 Message Date
evonneng 98a74c67de update lint error 2022-05-11 00:09:33 +00:00
evonneng a555ef7eb4 cap history in calc avg of stats; tensorboard output to /tmp (#67)
* cap history in calc avg of stats

* fixed too many nested ifs
2022-05-10 16:55:16 -07:00
Ethan Weber 5fb3d14304 Fixed testing and added graph abstractions (#64)
* abstracted some test logic and actually using test images now

* moved logging from trainer to graph

* fixing dataset path

* refactored instant ngp code to use inheritance

* fixed my run_train

* fixed data path

* fixed data dir
2022-05-09 11:54:40 -07:00
Matt Tancik 0e77c731e2 missing file 2022-05-09 11:16:04 -07:00
Matt Tancik b4e45d6fc0 Improve depth vis 2022-05-09 11:11:00 -07:00
evonneng a498a6cc64 Standardized logging code to all global var; writers now accessible from all files (#62)
* 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

* merge

* standardized all logging to global variables
2022-05-06 21:55:33 -07:00
Matt Tancik cf9c557b7b Add colormaps to logs 2022-05-06 18:07:12 -07:00
Ethan Weber 0b82c1188c Colmap type of dataset (from "friends" project) (#56)
* code to process the friends data. need to add masking

* code to process the friends data. need to add masking

* black

* trying to fix black

* black fixed

* updating black version
2022-05-05 08:45:30 -07:00
Ethan Weber 387ad93027 merging 2022-04-29 22:10:36 +00:00
evonneng a93dace958 modularizing hydra config and setting defaults (#46)
* lint

* reconfigured modules in yaml
2022-04-29 14:39:43 -07:00
Ethan Weber 7b118750ec merged master into this 2022-04-29 21:37:10 +00:00
Ethan Weber 6b17ad51c6 cpu support and multi gpu progress 2022-04-29 21:29:13 +00: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 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 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
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
Ethan Weber 80ad17700c pylint and test cases for dataloader 2022-04-23 19:45: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
Ethan Weber 5099a4f7c5 resizing image 2022-04-22 18:19:39 +00:00
Ethan Weber f090d9a001 training loop with tensorboard funs 2022-04-22 03:52:35 +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
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
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
evonneng bf3b77f952 Eng/distributed training (#27)
* added checkpointing logic

* small nits
2022-04-20 12:06:29 -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
evonneng 89a7d1ba53 added base writer class and tensorboard support; todo: init from config in trainer.py 2022-04-19 07:02:31 +00:00
Ethan Weber 0b7dfcdc5d Initial ray generation and dataset code (#12)
* ray generator and cameras structure progress

* test renames

* added pass

* conflicts

* fixing some black errors

* reformated files

* merged master and .gitignore

* fixing linting

* removing old folder

* removed return False

* pylint fixes

* update

* updates

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update
2022-04-12 15:08:16 -07:00
Matt Tancik 81aa20a3d5 add empty folders 2022-04-05 13:50:51 -07:00