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

fix dropout bug in PaddleVideo (#2860)

上级 ac7907e9
......@@ -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.
先完成此消息的编辑!
想要评论请 注册