PyTorch Toolbelt 0.0.5

BloodAxe released this 26 Apr 15:30

· 1120 commits to develop since this release

a03b276

Changes

Added 10-Crop TTA (#4)
Added unit tests for TTA functions
Added freeze_bn function to freeze all BN layers in a model
Rename unpad_tensor to unpad_image_tensor to mimick pad_image_tensor

Bugfixes

Fixed bug in d4_image2mask

Assets 2