如何保存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就可以保存了