Hi , Would you release your experiments environment?

sangwoomo / FreezeD

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

https://arxiv.org/abs/2002.10964

285 stars 37 forks source link

Hi , Would you release your experiments environment? #1

Closed Johnson-yue closed 4 years ago

Johnson-yue commented 4 years ago

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???

Johnson-yue commented 4 years ago

Sorry this is environment bug, and I fixed with:

torch : 1.2.0 torchvision : 0.3.0 cuda 10

sangwoomo commented 4 years ago

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

LendelTheGreat commented 4 years ago

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

Johnson-yue commented 4 years ago

@LendelTheGreat Thank you

sangwoomo commented 4 years ago

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 !