Bert-Chinese-Text-Classification-Pytorch icon indicating copy to clipboard operation
Bert-Chinese-Text-Classification-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon 649453932

模型存储为ckpt格式之后 怎么读取呢？需要把model的class重新写出来吗？

Open sense1011 opened this issue 4 years ago • 0 comments

刚开始用torch，已经参照您的方法自己训练了一个二分类模型，但我想把训练好的模型放在一个虚拟机上用，请问是直接load读取ckpt文件就可以了吗？

sense1011 avatar Jul 28 '21 07:07 sense1011