fix dropout bug in PaddleVideo (#2860) (#2879)

2e383363 · ruri · GitHub · b577bd7b · 2e383363
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleCV/PaddleVideo/models/nonlocal_model/resnet_video.py PaddleCV/PaddleVideo/models/nonlocal_model/resnet_video.py +1 -1

未找到文件。
--- a/PaddleCV/PaddleVideo/models/nonlocal_model/resnet_video.py
+++ b/PaddleCV/PaddleVideo/models/nonlocal_model/resnet_video.py
@@ -321,7 +321,7 @@ def create_model(data, label, cfg, is_training=True, mode='train'):
        pool_padding=[0, 0, 0],
        name='pool5')

-    if (cfg.TRAIN.dropout_rate > 0) and (test_mode is False):
+    if (cfg.TRAIN.dropout_rate > 0):
        blob_out = fluid.layers.dropout(
            blob_out, cfg.TRAIN.dropout_rate, is_test=test_mode)