inuex35

Please see the latest commit below. I'm using a 16GB GPU, and while it was running out of memory with 8K images, it seems to be working with 5K images....

Maybe the problem is here. The detected CUDA version (11.6) mismatches the version that was used to compile PyTorch (12.1). Please make sure to use the same CUDA versions. Check...

Im going to run this with both perspective and equirectangular image set together. When I confirm it to work, I will open a PR!