trocr-chinese RuntimeError: CUDA error: device-side assert triggered

RuntimeError: CUDA error: device-side assert triggered

Open wuxiaolianggit opened this issue 2 years ago • 3 comments

大佬，您好，我按照步骤来训练，但是出现这个错误，应该时类别不匹配导致的，这个该怎么解决呢 @wenlihaoyu @wenyinlong

Apr 26 '22 06:04 wuxiaolianggit

建议用readme提供的docker测试一下，可能是你这边各个包的依赖性造成。最好在linux环境下去训练

Apr 29 '22 03:04 wenyinlong

@wuxiaolianggit @wuxiaolianggit 你好你好，请问你的问题解决了吗？我也遇到同样的问题，我跑的是手写文本数据集。的用少数训练集可以跑通，换了大的数据集就出现这个问题了。求助

Jun 29 '23 05:06 Authen223

@wuxiaolianggit @Authen223 #43

Sep 24 '23 11:09 fangshuiyun