2023年3月22日

:)关于transformers模型的保存与加载-|

摘要: 关于transformers模型的保存与加载 两种情况, 自定义模型训练后保存, transformers预训练模型保存。 参考代码 # -*- coding: utf-8 -*- import torch from transformers import GPT2LMHeadModel from 阅读全文

posted @ 2023-03-22 21:58 lexn 阅读(1773) 评论(0) 推荐(0) 编辑

导航