Skip to content

Compatibility with post-Pytorch v1.0.0 BN state dicts

Compare
Choose a tag to compare
@ducksoup ducksoup released this 14 Aug 08:29

State dicts from standard BatchNorm layers trained with Pytorch v1.0.0 or newer can now be properly loaded by ABN, InPlaceABN and InPlaceABNSync.