Closed phdenzel closed 4 months ago
Default of res_groups = 32 which is used for GroupNorm. When in_channels < 32 there is an error, since res_norm > in_channels.
I suggest warning the user and automatically adjusting res_norm, e.g. with res_norm = min(in_channels, res_norm).
res_norm = min(in_channels, res_norm)
Default of res_groups = 32 which is used for GroupNorm. When in_channels < 32 there is an error, since res_norm > in_channels.
I suggest warning the user and automatically adjusting res_norm, e.g. with
res_norm = min(in_channels, res_norm)
.