Browse Source

update training config

main
bluestyle97 9 months ago
parent
commit
502613ce17
  1. 11
      configs/instant-mesh-large-train.yaml
  2. 9
      configs/instant-nerf-large-train.yaml

11
configs/instant-mesh-large-train.yaml

@ -9,17 +9,16 @@ model:
lrm_generator_config:
target: src.models.lrm_mesh.InstantMesh
params:
camera_embed_dim: 1024
rendering_samples_per_ray: 128
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
encoder_feat_dim: 768
encoder_freeze: false
encoder_model_name: facebook/dino-vitb16
rendering_samples_per_ray: 128
grid_res: 128
grid_scale: 2.1
@ -36,7 +35,7 @@ data:
meta_fname: filtered_obj_name.json
input_image_dir: rendering_random_32views
target_image_dir: rendering_random_32views
input_view_num: 6
input_view_num: 4
target_view_num: 4
total_view_n: 32
fov: 50

9
configs/instant-nerf-large-train.yaml

@ -8,17 +8,16 @@ model:
lrm_generator_config:
target: src.models.lrm.InstantNeRF
params:
camera_embed_dim: 1024
rendering_samples_per_ray: 128
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
encoder_feat_dim: 768
encoder_freeze: false
encoder_model_name: facebook/dino-vitb16
rendering_samples_per_ray: 128
data:

Loading…
Cancel
Save