sangwoomo / FreezeD

Freeze the Discriminator: a Simple Baseline for Fine-Tuning GANs (CVPRW 2020)

Home Page:https://arxiv.org/abs/2002.10964

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Hi , Would you release your experiments environment?

Johnson-yue opened this issue · comments

I am using, python3.7.3, cuda 10.0, cudnn 7.6.1 and then I run finetune.py as command:
python finetune.py --name AnimalFace_finetune --mixing --loss r1 --sched --dataset AnimalFace

and the Bug is :

cuDNN error: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a non-contiguous input

bug occur in line 312 in finetune.py

Do you understand this bug meaning???

Sorry this is environment bug, and I fixed with:

torch : 1.2.0
torchvision : 0.3.0
cuda 10

Hi, sorry for late response! Glad to hear that you fixed it! :)

Ran into the same issue. Managed to get it working with a nightly pytorch build:

torch: 1.5.0
torchvision: 0.6.0
cuda 9.1

Hi, sorry for the late response!
I remember that I had the same issue, but I forgot it since I fixed it a while ago.

It is a bug in PyTorch 1.4.0 (when one using .transpose() or .permute()), and one can fix it by upgrading to nightly version or downgrading. Thanks for the reminder @LendelTheGreat !