Akshay Kulkarni

Hi, did you try explicitly converting the variable to float using `.float()`? That usually works but I'm not sure where exactly you got the error.

The error is coming from the fastai dataset code that we had used. Could you check the fastai, pytorch and python versions that you are using? We did not face...

Could you try with PyTorch 1.3.1? we had used that version. You can use our [environment.yml](https://github.com/IvLabs/stagewise-knowledge-distillation/blob/master/environment.yml) file for installing a new environment. The reason is that PyTorch has had many...

Sorry, I don't have access to the original environment that I used and I cannot find the version of torchvision corresponding to 1.3.1 of torch. But you can try installing...

Closing due to inactivity. Please feel free to re-open or open another issue for other queries.