Imported from https://github.com/jotaf98/pytorch-curveball