21 Commits

Include renames

Author	SHA1	Message	Date
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
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