From 502613ce17a463f339811b117d586d51e869e153 Mon Sep 17 00:00:00 2001 From: bluestyle97 Date: Fri, 12 Apr 2024 17:10:31 +0800 Subject: [PATCH] update training config --- configs/instant-mesh-large-train.yaml | 11 +++++------ configs/instant-nerf-large-train.yaml | 9 ++++----- 2 files changed, 9 insertions(+), 11 deletions(-) diff --git a/configs/instant-mesh-large-train.yaml b/configs/instant-mesh-large-train.yaml index b5d99a2..06ffcbe 100644 --- a/configs/instant-mesh-large-train.yaml +++ b/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 diff --git a/configs/instant-nerf-large-train.yaml b/configs/instant-nerf-large-train.yaml index 3544f8e..0e0cc76 100644 --- a/configs/instant-nerf-large-train.yaml +++ b/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: