Skip to content

PyTorch Toolbelt 0.5.2

Compare
Choose a tag to compare
@BloodAxe BloodAxe released this 26 Aug 14:59
· 300 commits to master since this release
46f189f

BugFixes

  • Fixed bug in ApplySoftmaxTo and ApplySigmoidTo modules that could lead to activations not applied to input when it was a string

New API

  • Added fs.find_images_in_dir_recursive
  • Added utils.describe_outputs to return a human-friendly representation of complex (dict, nested list, etc) outputs to see shape, mean/std of each tensor.

Other

More MyPy fixes & type annotations