未验证 提交 2e383363 编写于 作者: R ruri 提交者: GitHub

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

上级 b577bd7b
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册