| 1 | from enum import Enum |
| 2 | |
| 3 | |
| 4 | class NormLayerType(Enum): |
| 5 | GROUP_NORM = "group_norm" |
| 6 | PIXEL_NORM = "pixel_norm" |
| 7 | |
| 8 | |
| 9 | class LogVarianceType(Enum): |
| 10 | PER_CHANNEL = "per_channel" |
| 11 | UNIFORM = "uniform" |
| 12 | CONSTANT = "constant" |
| 13 | NONE = "none" |
| 14 | |
| 15 | |
| 16 | class PaddingModeType(Enum): |
| 17 | ZEROS = "zeros" |
| 18 | REFLECT = "reflect" |
| 19 | REPLICATE = "replicate" |
| 20 | CIRCULAR = "circular" |
| 21 |