Skip to content

Split iABN library from training scripts, improved native code and synchronized BN
Compare
Choose a tag to compare
@ducksoup ducksoup released this 04 Jul 13:44
v1.0.0
a554a80

This release marks some major changes in inplace_abn:

  • Complete rewrite of the CUDA code following the most recent native BN implementation from Pytorch
  • Improved synchronized BN implementation, correctly handling different per-GPU batch sizes and Pytorch distributed groups
  • The iABN layers are now packaged in an installable python library to simplify use in other projects
  • The Imagenet / Vistas scripts are still available in the scripts folder
Assets 2
Loading