PyTorch Toolbelt 0.5.2
BugFixes
- Fixed bug in
ApplySoftmaxTo
andApplySigmoidTo
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