From 4619f414e43e6fad5b0cd5bc31e2fdc783b8e975 Mon Sep 17 00:00:00 2001 From: Guanghua Yu <742925032@qq.com> Date: Mon, 5 Jul 2021 15:26:17 +0800 Subject: [PATCH] fix export model with solov2_enhance (#3590) --- ppdet/modeling/heads/solov2_head.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ppdet/modeling/heads/solov2_head.py b/ppdet/modeling/heads/solov2_head.py index 36c120cf4..6af68bf61 100644 --- a/ppdet/modeling/heads/solov2_head.py +++ b/ppdet/modeling/heads/solov2_head.py @@ -252,7 +252,7 @@ class SOLOv2Head(nn.Layer): bias_attr=ParamAttr(initializer=Constant( value=float(-np.log((1 - 0.01) / 0.01)))))) - if self.drop_block: + if self.drop_block and self.training: self.drop_block_fun = DropBlock( block_size=3, keep_prob=0.9, name='solo_cate.dropblock') @@ -324,13 +324,13 @@ class SOLOv2Head(nn.Layer): for kernel_layer in self.kernel_pred_convs: kernel_feat = F.relu(kernel_layer(kernel_feat)) - if self.drop_block: + if self.drop_block and self.training: kernel_feat = self.drop_block_fun(kernel_feat) kernel_pred = self.solo_kernel(kernel_feat) # cate branch for cate_layer in self.cate_pred_convs: cate_feat = F.relu(cate_layer(cate_feat)) - if self.drop_block: + if self.drop_block and self.training: cate_feat = self.drop_block_fun(cate_feat) cate_pred = self.solo_cate(cate_feat) -- GitLab