Commit Graph
15 Commits
Author SHA1 Message Date
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 6b17ad51c6 cpu support and multi gpu progress 2022-04-29 21:29:13 +00: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 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 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 cf7034f0d1 added test set (using training data for now) 2022-04-22 03:45:06 +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