(module: nn.Module)
| 7 | |
| 8 | |
| 9 | def weight_init(module: nn.Module) -> None: |
| 10 | nn.init.kaiming_normal_(module.weight, mode="fan_out", nonlinearity="relu") |
| 11 | if module.bias is not None: # pyre-ignore |
| 12 | nn.init.constant_(module.bias, 0) |
| 13 | |
| 14 | |
| 15 | class Obj_Attn_Block(Module): |