mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

More appropriate name for se channel

Browse Source
This commit is contained in:
Ross Wightman 2019-05-10 23:32:18 -07:00
parent db056d97e2
commit 6523e4abe4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
models/genmobilenet.py
View File

@ -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