(name="silu", inplace=True)
| 15 | |
| 16 | |
| 17 | def get_activation(name="silu", inplace=True): |
| 18 | if name == "silu": |
| 19 | module = nn.SiLU(inplace=inplace) |
| 20 | elif name == "relu": |
| 21 | module = nn.ReLU(inplace=inplace) |
| 22 | elif name == "lrelu": |
| 23 | module = nn.LeakyReLU(0.1, inplace=inplace) |
| 24 | else: |
| 25 | raise AttributeError("Unsupported act type: {}".format(name)) |
| 26 | return module |
| 27 | |
| 28 | |
| 29 | class BaseConv(nn.Module): |