crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

BigGait_v1

Browse Source
This commit is contained in:
Bugjudger
2024-05-11 09:32:24 +08:00
parent a9f711d2e6
commit d7ce9c498d
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/biggait/BigGait_CCPG.yaml
+2 -4
View File
@@ -11,8 +11,7 @@ data_cfg:
evaluator_cfg:
enable_float16: true
restore_ckpt_strict: True
# restore_hint: 40000
restore_hint: /home/ydq/workspace/HID/Release/BigGait_MSU/output/CCPG/BigGait__Dinov2_Gaitbase/BigGait__Dinov2_Gaitbase_Frame30/checkpoints/BigGait__Dinov2_Gaitbase_Frame30-40000.pt
restore_hint: 40000
save_name: BigGait__Dinov2_Gaitbase_Frame30
eval_func: evaluate_CCPG
sampler:
@@ -38,8 +37,7 @@ loss_cfg:
model_cfg:
model: BigGait__Dinov2_Gaitbase
pretrained_dinov2: /home/ydq/workspace/HID/Release/BigGait_MSU/pretrained_LVMs/dinov2_vits14_pretrain.pth # pretrain DINOv2
# pretrained_dinov2: pretrained_LVMs/dinov2_vits14_pretrain.pth # pretrain DINOv2
pretrained_dinov2: pretrained_LVMs/dinov2_vits14_pretrain.pth # pretrain DINOv2
pretrained_mask_branch: None
# pretrained_mask_branch: pretrained_LVMs/MaskBranch_vits14.pt # load pretrain Mask_Branch
image_size: 224 # 448x224