From 2e3833634ef9438cda15c2459194bd3152e3e243 Mon Sep 17 00:00:00 2001 From: ruri Date: Mon, 22 Jul 2019 15:28:30 +0800 Subject: [PATCH] fix dropout bug in PaddleVideo (#2860) (#2879) --- PaddleCV/PaddleVideo/models/nonlocal_model/resnet_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PaddleCV/PaddleVideo/models/nonlocal_model/resnet_video.py b/PaddleCV/PaddleVideo/models/nonlocal_model/resnet_video.py index 6a071756..5fa53be3 100644 --- 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) -- GitLab