Closed aiXander closed 4 years ago
Ok, turns out this error is actually caused by setting sched.minibatch_gpu_base = 1 in run_training.py. At first I was getting OOM errors, so I lowered the gpu_base, but if you set it to 1 you suddenly get this weird and uninformative error..
Hey absolutely love the functionalities of this repo!! Thx so much!
However, I'm getting issues when running this on my local RTX 2080 Ti machine.. All the training ops run fine, except for the _G_regop
The weird thing is that I just tested with the code from the official StyleGAN2 repo, and there everything seems to work perfectly (including G_reg_op)...
To summarize, the core of the error seems to originate in the custom cuda functions for upsampling:![Screenshot from 2020-02-07 01-00-37](https://user-images.githubusercontent.com/24993604/73989229-6f791a00-4945-11ea-85fd-a87e9451aa9e.png)
Here's the full traceback: