Incompatibility with amd gpu's

[WaterGazes]

Since TensorFlow uses cuda cores this module isn't compatible with any amd GPU which would of been nice to know in the README. Another thing to note is your code to fix the torch.load() issue has not been updated on pypi. Something else you might want to add on the readme is that if you are on Linux with an amd GPU, amd has developed a tool called ROCm to allow tensorflow to use an amd's GPU cores.