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

Open muxiaoq opened this issue 2 years ago • 1 comments

作者你好，我在尝试复现代码。第一个模型复现成功，但是在使用第二个模型pretrain model without linear shaoe augmentation and non-linear shape augmentation here时，出现如下错误

Screenshot from 2023-12-06 11-44-35 这种情况应该如何处理，让代码成功复现？我使用的显卡是单卡3090

谢谢

Dec 06 '23 03:12 muxiaoq

I have the same problem when loading the pth, what should I do?

Jan 04 '24 10:01 Ironman6666