(
kernel_size: Union[int, Tuple[int, ...]],
)
| 64 | |
| 65 | |
| 66 | def get_same_padding( |
| 67 | kernel_size: Union[int, Tuple[int, ...]], |
| 68 | ) -> Union[int, Tuple[int, ...]]: |
| 69 | if isinstance(kernel_size, tuple): |
| 70 | return tuple([get_same_padding(ks) for ks in kernel_size]) |
| 71 | else: |
| 72 | assert kernel_size % 2 > 0, f"kernel size {kernel_size} should be odd number" |
| 73 | return kernel_size // 2 |
| 74 | |
| 75 | |
| 76 | class ConvLayer(nn.Module): |