GAIN

GAIN copied to clipboard

Flowers and IP102 datasets

2

Hi, could you upload the datasets maybe somewhere ? Thanks.

ilyak93

Runing without Cuda

2

What do I have to change in the code to run the model without cuda, just on cpu (is it possible?) ? I have downloaded the Apex version for only...

ilyak93

can u give us a tutorial that how to use this code？ for example. how to use dockerfile? And which dataset can we use to train model

Amosvcc

with torch.enable_grad()

1

Hi, in forward( ) function, in gain.py if remove the code " with torch.enable_grad():" what will it happen? Does it influence the final result?

gk966988

Bug report

1

At /src/models/ gain.py 86 gradient = logits * labels_ohe 87 grad_logits = (logits * labels_ohe).sum() 88 grad_logits.backward(gradient=gradient, retain_graph=True) Line 88 shows an error: File "C:\ProgramData\Anaconda2\envs\python3\lib\site-packages\torch\autograd\__init__.py", line 90, in backward allow_unreachable=True)...

guopengf

Why use class AdamW instead of "torch.optim.adamw "

2

Hi, Thanks a lot for sharing the pytorch version of GAIN!! I'm trying to run the code and make some changes based on GAIN. I was wondering why you define...

Dawn90