Closed kwhuang88228 closed 2 years ago
Hi, Thanks for reaching out!
I recommend in the following line: https://github.com/dorarad/gansformer/blob/main/dnnlib/tflib/custom_ops.py#L130 try changing int(tf_ver < 1.15) to 0.
Then you should clean the custom ops built so that you can retry:
rm -rf /external_code/gan/gansformer/dnnlib/tflib/cudacache/
and then try to run the code again. See the following issues (https://github.com/dorarad/gansformer/issues/7, https://github.com/dorarad/gansformer/issues/8) for further discussion and let me know if the solution works!
Thanks Drew! Deleting the cuda cache did the trick
Awesome! :-)
Hi Drew, I'm getting the following error both when I train a GANformer model on the clevr dataset from scratch or when I fine-tune a pretrained model. I didn't have this issue before the repo was updated with PyTorch implementation. I've also tried this and this without luck. Do you have any ideas?
Environment: Python 3.6.13 tensorflow-gpu 1.14.0 CUDA 9.1 cudnn 7