39 Commits

Author	SHA1	Message	Date
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
Matt Tancik	8c1e9d7f1a	Better align with nerf	2022-05-10 10:18:20 -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
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
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
Matt Tancik	fc282f8857	Add multiple color options	2022-05-04 15:47:54 -07:00
Matt Tancik	ba28d0cdb2	Add training with white backgrounds	2022-05-04 13:45:53 -07:00
evonneng	603c85ccd2	fixed error; changed pkg to file	2022-05-03 06:26:11 +00:00
evonneng	35145cb400	moved test config to own config dir; using default	2022-05-03 06:10:34 +00:00
Matt Tancik	a10eb5caef	Add basic NGP encoder	2022-05-02 13:27:30 -07:00
evonneng	e874e856ec	changed test config file to inherit from default (#49) ... * lint * reconfigured modules in yaml * added data module * changed test config file	2022-04-29 16:18:30 -07:00
Ethan Weber	023e6a3f92	merging masters new config	2022-04-29 22:29:43 +00:00
Ethan Weber	92062418d6	Merge branch 'master' into ethan/apr29_cpu_and_multigpu	2022-04-29 22:10:51 +00:00
Ethan Weber	387ad93027	merging	2022-04-29 22:10:36 +00:00
evonneng	9004584026	added data/ config folder (#47) ... * lint * reconfigured modules in yaml * added data module	2022-04-29 15:01:39 -07: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
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