AaaJrwp4

Hey @jjdredd , thank you so much for your code! I am trying to use it and got an error. Python 3.8 Cuda 11.7 Torch 1.10.0+cu111, Torchvision 0.11.0, Torchaudio 0.10.0...