search

lkpengcs / DCDA

MIT License
14 stars 3 forks source link

RuntimeError #2

Open manigalati opened 1 year ago

manigalati commented 1 year ago

Good morning, When running your model, during the second for loops I get the following error (line 78 in train.py).

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [256, 8]], which is output 0 of AsStridedBackward0, is at version 2; expected version 1 instead.

Do you have any idea of the why? Thanks in advance :)

lkpengcs commented 1 year ago

Could you please provide more details? I do not have the file 'main.py'.

manigalati commented 1 year ago

Sorry, I have updated my issue. I meant "train.py" not "main.py".

lkpengcs commented 1 year ago

Maybe you can try PyTorch 1.4