mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix an an untested change, remove a debug print

pull/155/head
Ross Wightman 2020-06-01 17:26:42 -07:00
parent eb7653614f
commit f28170df3f
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/efficientnet_blocks.py
View File

@ -220,7 +220,7 @@ class InvertedResidual(nn.Module):
has_se = se_ratio is not None and se_ratio > 0. has_se = se_ratio is not None and se_ratio > 0.
self.has_residual = (in_chs == out_chs and stride == 1) and not noskip self.has_residual = (in_chs == out_chs and stride == 1) and not noskip
self.drop_path_rate = drop_path_rate self.drop_path_rate = drop_path_rate
print(act_layer)
# Point-wise expansion # Point-wise expansion
self.conv_pw = create_conv2d(in_chs, mid_chs, exp_kernel_size, padding=pad_type, **conv_kwargs) self.conv_pw = create_conv2d(in_chs, mid_chs, exp_kernel_size, padding=pad_type, **conv_kwargs)
self.bn1 = norm_layer(mid_chs, **norm_kwargs) self.bn1 = norm_layer(mid_chs, **norm_kwargs)

2
timm/models/layers/norm_act.py
View File

@ -60,7 +60,7 @@ class BatchNormAct2d(nn.BatchNorm2d):
if torch.jit.is_scripting(): if torch.jit.is_scripting():
x = self._forward_jit(x) x = self._forward_jit(x)
else: else:
self._forward_python(x) x = self._forward_python(x)
if self.act is not None: if self.act is not None:
x = self.act(x) x = self.act(x)
return x return x