File size: 1,672 Bytes
2fe3da0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
model:
base_learning_rate: 4.0e-06
target: src.model_mesh.MVRecon
params:
mesh_save_root: Objaverse
init_ckpt: nerf_base.ckpt
input_size: 512
render_size: 512
use_tv_loss: true
sample_points: null
use_gt_albedo: false
lrm_generator_config:
target: src.models.lrm_mesh.PRM
params:
encoder_feat_dim: 768
encoder_freeze: false
encoder_model_name: facebook/dino-vitb16
transformer_dim: 1024
transformer_layers: 16
transformer_heads: 16
triplane_low_res: 32
triplane_high_res: 64
triplane_dim: 80
rendering_samples_per_ray: 128
grid_res: 128
grid_scale: 2.1
data:
target: src.data.objaverse.DataModuleFromConfig
params:
batch_size: 1
num_workers: 8
train:
target: src.data.objaverse.ObjaverseData
params:
root_dir: Objaverse
light_dir: env_mipmap
input_view_num: [6]
target_view_num: 6
total_view_n: 18
distance: 5.0
fov: 30
camera_random: true
validation: false
validation:
target: src.data.objaverse.ValidationData
params:
root_dir: Objaverse
input_view_num: 6
input_image_size: 320
fov: 30
lightning:
modelcheckpoint:
params:
every_n_train_steps: 100
save_top_k: -1
save_last: true
callbacks: {}
trainer:
benchmark: true
max_epochs: -1
val_check_interval: 2000000000
num_sanity_val_steps: 0
accumulate_grad_batches: 8
log_every_n_steps: 1
check_val_every_n_epoch: null # if not set this, validation does not run
|