Update test_animal3d.sh to modify eval_sample_steps, change model_type, and update saved_model_path for consistency with model registry usage.

Author: xiu-cs

animals/scripts/test_animal3d.sh

@@ -2,25 +2,26 @@ layers=5
 batch_size=13
 lr=1e-3
 gpu_id=0
-eval_sample_steps=3
+eval_sample_steps=5
 num_saved_models=3
 frames=1
 large_decay_epoch=15
 lr_decay_large=0.75
 n_joints=26
 out_joints=26
 epochs=300
-# model_path='models/model_animals.py'
-model_path='./pre_trained_models/animal3d_pretrained_weights/model_animal3d.py' # when the path is empty, the model will be loaded from the installed fmpose package
-saved_model_path='./pre_trained_models/animal3d_pretrained_weights/CFM_154_4403_best.pth'
+model_type='fmpose3d_animals'
+# model_path='' # set to a local file path to override the registry
+saved_model_path='./pre_trained_models/fmpose3d_animals/fmpose3d_animals_pretrained_weights.pth'
+
 # root path denotes the path to the original dataset
 root_path="./dataset/"
 train_dataset_paths=(
   "./dataset/animal3d/train.json"
   "./dataset/control_animal3dlatest/train.json"
 )
 test_dataset_paths=(
-  "./dataset/control_animal3dlatest/test.json"
+  "./dataset/animal3d/test.json"
 )
 
 folder_name="TestCtrlAni3D_L${layers}_lr${lr}_B${batch_size}_$(date +%Y%m%d_%H%M%S)"
@@ -33,6 +34,7 @@ python ./scripts/main_animal3d.py \
   --test 1 \
   --batch_size ${batch_size} \
   --lr ${lr} \
+  --model_type "${model_type}" \
   ${model_path:+--model_path "$model_path"} \
   --folder_name ${folder_name} \
   --layers ${layers} \