mmammoth

A problem about:"RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation"

1

Excuse , I have a question, when I run it, I faced a problem like this: "**RuntimeError: one of the variables needed for gradient computation has been modified by an...

使用自己的数据集训练失败的问题？

1

train.py中把batch_size调小之后还是会出现“CUDA out of memory. Tried to allocate 4.16 GiB”的问题，请问是哪里需要调整呢？