问答交流

如何保存Transformer或者torch的模型参数?

由howie1013创建,最终由small_q 被浏览 23 用户

问题

我用torch.save会报错,

直接DataSource.write_pickle(model),会提示

<trackeback: AttributeError: Can't pickle local object 'get_cosine_schedule_with_warmup.<locals>.lr_lambda'>

解答

已经解决了,model.schedulers = None就可以保存了

标签

AttributeErrorTransformer
评论
  • 已经解决了,model.schedulers = None就可以保存了
{link}