mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 10:34:29 +08:00
676beebb14
* some initial cuda code as I learn. using git to save progress. * first cuda kernel for ray sampling that mostly works * cleaning up code. working on lr. structs for scene bounds aabb * added scene bounds collider code * updated setup * setup * setup * black * including some files * exporting env variable * new file * git workflows * github actions * github actions * github actions * removing env var
41 lines
922 B
YAML
41 lines
922 B
YAML
# saving model specifications
|
|
model_dir: /tmp/mattport_models/
|
|
steps_per_save: null
|
|
steps_per_test: 100
|
|
max_num_iterations: 1000000
|
|
|
|
# pretrained loading options
|
|
resume_train:
|
|
load_dir: # pretrained model directory
|
|
load_step: # pretrained model's step number
|
|
|
|
network:
|
|
_target_: mattport.nerf.graph.vanilla_nerf.NeRFGraph
|
|
near_plane: 2.0
|
|
far_plane: 6.0
|
|
num_coarse_samples: 64
|
|
num_importance_samples: 128
|
|
|
|
loss_coefficients: # loss coefficients
|
|
rgb_loss_coarse: 1.0
|
|
rgb_loss_fine: 1.0
|
|
|
|
# optimizer options for network
|
|
param_groups:
|
|
fields:
|
|
optimizer:
|
|
_target_: torch.optim.Adam
|
|
lr: 0.0005
|
|
scheduler:
|
|
_target_: mattport.nerf.optimizers.ExponentialDecaySchedule
|
|
lr_final: 0.000005
|
|
max_steps: 1000000
|
|
|
|
cameras:
|
|
optimizer:
|
|
_target_: torch.optim.SGD
|
|
lr: 1e-4
|
|
scheduler:
|
|
_target_: torch.optim.lr_scheduler.ExponentialLR
|
|
gamma: 0.1
|