More appropriate name for se channel

pull/1/head
Ross Wightman 6 years ago
parent db056d97e2
commit 6523e4abe4

@ -549,8 +549,8 @@ class InvertedResidual(nn.Module):
# Squeeze-and-excitation
if self.has_se:
reduce_mult = mid_chs if se_reduce_mid else in_chs
self.se = SqueezeExcite(mid_chs, reduce_chs=max(1, int(reduce_mult * se_ratio)),
se_base_chs = mid_chs if se_reduce_mid else in_chs
self.se = SqueezeExcite(mid_chs, reduce_chs=max(1, int(se_base_chs * se_ratio)),
act_fn=act_fn, gate_fn=se_gate_fn)
# Point-wise linear projection

Loading…
Cancel
Save