solve the LF char problem

PaddlePaddle · May 12, 2021 · 059ac86 · 059ac86
1 parent 74128a6
commit 059ac86
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/paddlevideo/tasks/train.py b/paddlevideo/tasks/train.py
@@ -132,14 +132,13 @@ def train_model(cfg,
         model.train()
 
         # Freeze all BN layers except the first BN layer during finetune in ucf101
-        if weights:
+        if weights and :
             count = 0
             for m in model.sublayers():
                 if isinstance(m, paddle.nn.BatchNorm2D):
                     count += 1
                     if count >= 2:
                         m.eval()
-                        # shutdown update in frozen mode
                         m.weight.stop_gradient = True
                         m.bias.stop_gradient = True
 

diff --git a/run.sh b/run.sh
@@ -6,18 +6,18 @@ start_time=$(date +%s)
 # run tsm training
 python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3" --log_dir=log_tsm main.py  --validate -c configs/recognition/tsm/tsm_k400_frames.yaml
 
-# run tsm fine tune in ucf101 dataset
-#python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3" --log_dir=log_tsm main.py  --validate -c configs/recognition/tsm/tsm_ucf101_frames.yaml --weights /workspace/hesensen/Alignworks/PaddleVideo_TSM_finetune_ucf101_135/PaddleVideo_TSM_finetune_ucf101/TSM_best.pdparams
-
-# run tsm amp fine tune in ucf101 dataset,
-#python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3" --log_dir=log_tsm main.py  --amp --validate -c configs/recognition/tsm/tsm_ucf101_frames.yaml --weights /workspace/hesensen/Alignworks/PaddleVideo_TSM_finetune_ucf101_135/PaddleVideo_TSM_finetune_ucf101/TSM_best.pdparams
-
 # run tsm amp training
 #python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3" --log_dir=log_tsm main.py  --amp --validate -c configs/recognition/tsm/tsm_k400_frames.yaml
 
 # run tsm amp training, nhwc
 #python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3" --log_dir=log_tsm main.py  --amp --validate -c configs/recognition/tsm/tsm_nhwc.yaml
 
+# run tsm finetune in ucf101 dataset
+#python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3" --log_dir=log_tsm main.py  --validate -c configs/recognition/tsm/tsm_ucf101_frames.yaml --weights data/TSM_k400_best.pdparams
+
+# run tsm amp finetune in ucf101 dataset
+#python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3" --log_dir=log_tsm main.py  --amp --validate -c configs/recognition/tsm/tsm_ucf101_frames.yaml --weights data/TSM_k400_best.pdparams
+
 # run tsn training
 #python3.7 -B -m paddle.distributed.launch --gpus="0,1,2,3" --log_dir=log_tsn main.py  --validate -c configs/recognition/tsn/tsn.yaml