muxiaoq

使用第二个checkpoint模型复现代码时报错

1

作者你好，我在尝试复现代码。第一个模型复现成功，但是在使用第二个模型pretrain model without linear shaoe augmentation and non-linear shape augmentation [here](https://drive.google.com/file/d/1885sFjQz1v0SL5z92a-3KSZcf2zj5BHg/view?usp=drive_link)时，出现如下错误  这种情况应该如何处理，让代码成功复现？ 我使用的显卡是单卡3090 谢谢