今天在用libtorch训练得时候,想用预训练模型加速训练,居然报错,以为是模型得问题,然后重新训练了一个模型,作为预训练模型,还是报错,一时找不到原因,最后在大佬的帮助下搞定了,
先贴上错误的:
这是我以为的,
实际上应该这样:
这里有一些torch::load和from_load应该有很大差别,具体什么差别呢,等有时间在查,只知道前者用在部署的时候
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)