Fixed bugs for my pytorch like library
Well so I have might have broken the gpu version for okrolearn but now it is...
Tags: pytorch, python, tensorflow, machinelearning