热门
用8卡拉起来的训练,ckpt被分成很多个,如下, 目前官网看到恢复训练的代码是load .pth文件。
这么多个分片的参数文件如何加载?
您好,与文档中load模型类似,checkpoint_path 只需填写到 os.path.join(self.tmp_dir, 'epoch_1'),或是 iter_xxx 这一级即可,不用添加 .pth 后缀,读取过程后面会自动拼接文件名
包含命名实体识别、文本分类、分词、关系抽取、问答、推理、文本摘要、情感分析、机器翻译等多个领域