CAIIVS / chuchichaestli

Where you find all the state-of-the-art cooking utensils (salt, pepper, gradient descent... the usual).
GNU General Public License v3.0
3 stars 0 forks source link

Error when residual blocks in_channel < 32 #19

Closed phdenzel closed 4 months ago

phdenzel commented 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).