You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.6 KiB
67 lines
1.6 KiB
9 months ago
|
model:
|
||
|
|
base_learning_rate: 4.0e-04
|
||
|
|
target: src.model.MVRecon
|
||
|
|
params:
|
||
|
|
input_size: 320
|
||
|
|
render_size: 192
|
||
|
|
|
||
|
|
lrm_generator_config:
|
||
|
|
target: src.models.lrm.InstantNeRF
|
||
|
|
params:
|
||
|
|
camera_embed_dim: 1024
|
||
|
|
rendering_samples_per_ray: 128
|
||
|
|
transformer_dim: 1024
|
||
|
|
transformer_layers: 16
|
||
|
|
transformer_heads: 16
|
||
|
|
triplane_low_res: 32
|
||
|
|
triplane_high_res: 64
|
||
|
|
triplane_dim: 80
|
||
|
|
encoder_feat_dim: 768
|
||
|
|
encoder_freeze: false
|
||
|
|
encoder_model_name: facebook/dino-vitb16
|
||
|
|
|
||
|
|
|
||
|
|
data:
|
||
|
|
target: src.data.objaverse.DataModuleFromConfig
|
||
|
|
params:
|
||
|
|
batch_size: 2
|
||
|
|
num_workers: 8
|
||
|
|
train:
|
||
|
|
target: src.data.objaverse.ObjaverseData
|
||
|
|
params:
|
||
|
|
root_dir: data/objaverse
|
||
|
|
meta_fname: filtered_obj_name.json
|
||
|
|
input_image_dir: rendering_random_32views
|
||
|
|
target_image_dir: rendering_random_32views
|
||
|
|
input_view_num: 6
|
||
|
|
target_view_num: 4
|
||
|
|
total_view_n: 32
|
||
|
|
fov: 50
|
||
|
|
camera_rotation: true
|
||
|
|
validation: false
|
||
|
|
validation:
|
||
|
|
target: src.data.objaverse.ValidationData
|
||
|
|
params:
|
||
|
|
root_dir: data/valid_samples
|
||
|
|
input_view_num: 6
|
||
|
|
input_image_size: 320
|
||
|
|
fov: 30
|
||
|
|
|
||
|
|
|
||
|
|
lightning:
|
||
|
|
modelcheckpoint:
|
||
|
|
params:
|
||
|
|
every_n_train_steps: 1000
|
||
|
|
save_top_k: -1
|
||
|
|
save_last: true
|
||
|
|
callbacks: {}
|
||
|
|
|
||
|
|
trainer:
|
||
|
|
benchmark: true
|
||
|
|
max_epochs: -1
|
||
|
|
gradient_clip_val: 1.0
|
||
|
|
val_check_interval: 1000
|
||
|
|
num_sanity_val_steps: 0
|
||
|
|
accumulate_grad_batches: 1
|
||
|
|
check_val_every_n_epoch: null # if not set this, validation does not run
|