From f0c71f175044b20b6420f97dfc3496f37288d352 Mon Sep 17 00:00:00 2001
From: Dongyang Jin <73057174+jdyjjj@users.noreply.github.com>
Date: Sat, 25 May 2024 09:33:41 +0800
Subject: [PATCH] Update BigGait_CCPG.yaml

---
 configs/biggait/BigGait_CCPG.yaml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/configs/biggait/BigGait_CCPG.yaml b/configs/biggait/BigGait_CCPG.yaml
index 9ab1308..b20b3aa 100644
--- a/configs/biggait/BigGait_CCPG.yaml
+++ b/configs/biggait/BigGait_CCPG.yaml
@@ -38,8 +38,7 @@ loss_cfg:
 model_cfg:
   model: BigGait__Dinov2_Gaitbase
   pretrained_dinov2: ./pretrained_LVMs/dinov2_vits14_pretrain.pth                   # DINOv2 Download Link: https://dl.fbaipublicfiles.com/dinov2/dinov2_vits14/dinov2_vits14_pretrain.pth
-  pretrained_mask_branch: None
-  # pretrained_mask_branch: ./pretrained_LVMs/MaskBranch_vits14.pt                # load pretrain Mask_Branch
+  pretrained_mask_branch: ./pretrained_LVMs/MaskBranch_vits14.pt                # pretrained_mask_branch: None or MaskBranch Download Link: https://drive.google.com/drive/folders/1zrWPUsrbCpwxoLgfom3d2irgxkBqtXqc?usp=sharing
   image_size: 224                                                               # 448x224
   sils_size: 32                                                                 # 64x32