Update graphgym/custom_graphgym/act/example.py

Author: Padarn

graphgym/custom_graphgym/act/example.py

@@ -20,5 +20,5 @@ def forward(self, x):
             return x * torch.sigmoid(x)
 
 
-register_act('swish', lambda: SWISH(inplace=cfg.mem.inplace))
+register_act('swish', partial(SWISH, inplace=cfg.mem.inplace))
 register_act('lrelu_03', partial(nn.LeakyReLU, 0.3, inplace=cfg.mem.inplace))