https://discuss.pytorch.org/t/move-numpy-dataset-to-gpu-memory/55758 https://discuss.pytorch.org/t/how-to-use-collate-fn/27181/2 https://pytorch.org/docs/stable/_modules/torchvision/transforms/functional.html#five_crop
