Open dieuroi opened 3 years ago
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation
I think the backward operation in train_one_batch (line 308 in transformer.py) has freed the graph, while this does not disturb the out-loop backward.
The error occurs when batch_loss.backward() (line 167 in MAML.py)
same error, how to fix
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation
I think the backward operation in train_one_batch (line 308 in transformer.py) has freed the graph, while this does not disturb the out-loop backward.
The error occurs when batch_loss.backward() (line 167 in MAML.py)