4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Commit Graph

25 Commits

Author SHA1 Message Date
evonneng
9671ec3c33 update readme 2022-05-11 17:59:35 +00: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
evonneng
3a35bcda91
adding scripts for easy local run (#59) 2022-05-05 13:48:34 -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
023e6a3f92 merging masters new config 2022-04-29 22:29:43 +00: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
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
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
Ethan Weber
be6fbe799e most code for nerf running 2022-04-19 22:32:10 +00:00
evonneng
b8e2713643 Changed Graph to nn.ModuleDict; Added DDP logic to seup_graph; locallyrunning DDP 2022-04-15 23:35:00 +00:00
evonneng
62d4218421 updated logic within Trainer class to allow for multiprocess training 2022-04-15 21:53:35 +00:00
evonneng
ac3e7515ef done adding main logic in run_train_nerf to initialize processes with more robust handling of DDP 2022-04-15 21:33:50 +00:00
evonneng
7dc9d445b2 added basic distributed logic for trainer 2022-04-15 01:04:17 +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
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
b63d5854a1 reworked structure to use packages 2022-04-02 20:07:41 +00:00
Ethan Weber
6aae70df32 first commit 2022-03-29 17:26:54 +00:00