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
Ethan Weber
391803759e
adding config
2022-04-23 17:37:13 +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
b6f878f227
merge
2022-04-21 20:29:43 +00: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
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
af74b4e95c
check DDP running correctly
2022-04-21 17:06:29 +00:00
Ethan Weber
be6fbe799e
most code for nerf running
2022-04-19 22:32:10 +00:00
evonneng
1aed15faad
changed config loading to hydra for graph generation + tests
2022-04-13 00:04:58 +00:00
evonneng
69ea3ae45e
rudimentary code; passing logic for ordering and input size calculation; need to check design decisions
2022-04-12 03:55:31 +00:00
evonneng
62aee54611
done logic to instantiate graph, test to check instantiation
2022-04-09 05:12:14 +00:00
Ethan Weber
91f57726ae
some initial module structuring
2022-04-08 17:01:47 +00:00
Ethan Weber
5691d00d42
fixed most linting errors
2022-04-07 20:22:48 +00:00
Ethan Weber
a7167f890e
brainstorming trainer code but broken
2022-04-06 22:45:53 +00:00
Ethan Weber
a9c0d5b97f
hydra config and starting trainer class
2022-04-06 16:46:47 +00:00
Ethan Weber
6aae70df32
first commit
2022-03-29 17:26:54 +00:00
evonneng
evonneng
Ethan Weber