Closed paganpasta closed 2 years ago
Currently, gradient computation is not supported with BatchNorm models. With the next release of Equinox, it would be nice to have test cases to check that networks are passing gradient computation.
Added in https://github.com/paganpasta/eqxvision/releases/tag/v0.1.4
Currently, gradient computation is not supported with BatchNorm models. With the next release of Equinox, it would be nice to have test cases to check that networks are passing gradient computation.