[Fix] GELU got an unexpected keyword argument 'inplace' (#1847)

a02a4815 · Rockey · GitHub · f2c50930 · a02a4815
Unverified Commit a02a4815 authored Apr 02, 2022 by Rockey Committed by GitHub Apr 02, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmcv/cnn/bricks/conv_module.py mmcv/cnn/bricks/conv_module.py +1 -1

No files found.
--- a/mmcv/cnn/bricks/conv_module.py
+++ b/mmcv/cnn/bricks/conv_module.py
@@ -157,7 +157,7 @@ class ConvModule(nn.Module):
            act_cfg_ = act_cfg.copy()
            # nn.Tanh has no 'inplace' argument
            if act_cfg_['type'] not in [
-                    'Tanh', 'PReLU', 'Sigmoid', 'HSigmoid', 'Swish'
+                    'Tanh', 'PReLU', 'Sigmoid', 'HSigmoid', 'Swish', 'GELU'
            ]:
                act_cfg_.setdefault('inplace', inplace)
            self.activate = build_activation_layer(act_cfg_)