fix multimae wrong class

HaoyiZhu · Aug 31, 2024 · 3e751d3 · 3e751d3
1 parent 069f109
commit 3e751d3
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/configs/exp_maniskill2_diffusion_policy/maniskill2_model/pretrained_multimae_rgbd.yaml b/configs/exp_maniskill2_diffusion_policy/maniskill2_model/pretrained_multimae_rgbd.yaml
@@ -25,5 +25,5 @@ model:
           type: low_dim
     obs_encoder:
       rgb_model:
-        _target_: src.models.components.img_encoder.multivit.MultiMAEModel
+        _target_: src.models.components.img_encoder.multivit.MultiViTModel
       use_depth: true
diff --git a/configs/exp_rlbench_act_policy/rlbench_model/pretrained_multimae_rgbd.yaml b/configs/exp_rlbench_act_policy/rlbench_model/pretrained_multimae_rgbd.yaml
@@ -16,4 +16,4 @@ trainer:
 model:
   policy:
     backbone: 
-      _target_: src.models.components.img_encoder.multivit.MultiMAEModel
+      _target_: src.models.components.img_encoder.multivit.MultiViTModel
diff --git a/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_multimae_rgbd.yaml b/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_multimae_rgbd.yaml
@@ -29,7 +29,6 @@ model:
           type: low_dim
     obs_encoder:
       rgb_model:
-        _target_: src.models.components.img_encoder.multivit.MultiMAEModel
+        _target_: src.models.components.img_encoder.multivit.MultiViTModel
         type: MultiViT
-        ckpt_path: /mnt/lustre/hetong/.multivit/multimae-b_98_rgb+-depth-semseg_1600e_multivit-afff3f8c.pth
       use_depth: true
diff --git a/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_ponderv2_pcd.yaml b/configs/exp_rlbench_diffusion_policy/rlbench_model/pretrained_ponderv2_pcd.yaml
@@ -29,7 +29,7 @@ model:
         _target_: src.models.components.pcd_encoder.spunet.SpUNet
         in_channels: 6
         num_classes: 0
-        pretrained_path: /mnt/lustre/share_data/ponder/pointcept/ponderv2-ppt-pretrain-scannet-s3dis-structured3d.pth
+        pretrained_path: checkpoints/ponderv2-ppt-pretrain-scannet-s3dis-structured3d.pth
       pcd_nsample: 16
       pcd_npoints: 2048
       use_mask: false