PyTorch Toolbelt 0.5.3

Bugfixes

• Fix #78 thanks https://github.com/mehran66 for pointing this out

New Stuff

• InriaAerialImageDataset for working with Inria Aerial Dataset
• get_collate_for_dataset function to get collate fn if a dataset instance (argument) exposes get_collate_fn method. Works also for ConcatDataset.

Improvements

• DatasetMeanStdCalculator supports dtype to specify accumulator type (float64 by default)