smxzhangxiaaobo

我那错误是自己写的损失类代码里，忘了加loss.data导致的，unimol本身代码没问题 > ***@***.***> > 时间： 2024年6月2日 (周日) 20:55 > 主题： Re: [dptech-corp/Uni-Mol] 往logging_output写日志时导致内存泄露问题 (Issue #173) > ***@***.***> > ***@***.***>, ***@***.***> > 请问你上述说的错误的代码是在哪个文件的第几行， 我现在训练蛋白-小分子复合体生成任务中的logging_output已经加了loss.data, 但是内存依旧随着训练轮次增加内存一直往上涨。请问你有和我一样的现象么 > — > Reply to this...