|
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 |
|
|
