Pytorch
pytorch语法
胤风
搞NLP的文青
展开
-
Pytorch之模型加载/保存
pytorch保存模型主要有两种方法:保存整个模型 (结构+参数)只保存参数(官方推荐)两者都是用torch.save(obj, dir)实现,这个函数的作用是将对象保存到磁盘中,它的内部其实也是使用Python的pickle实现。两种方法的区别是:保存整个模型的obj就直接是model对象,只保存参数的obj是从model里获取的存储了model参数的词典对象,前者太占内存,而且不太灵活,故官方推荐只保存参数,然后在建好模型的基础上加载。只保存参数...原创 2020-08-15 16:39:21 · 778 阅读 · 0 评论 -
手把手教你用fairseq训练一个NMT机器翻译系统
好的原创 2020-07-11 15:26:28 · 10270 阅读 · 32 评论 -
Pytorch tensor基础知识
tensor是深度学习计算中的基本数据结构,本文主要介绍Pytorch中的tensor对象的基础知识,包括tensor对象中的常用属性以及方法。Tensor属性Tensor对象,即torch.Tensor的大多属性都不是基本数据类型,而是Pytorch中定义的类,比如torch.dtype、torch.device等torch.dtypetensor.dtype 属性标识了 torch.T...原创 2020-04-30 23:19:11 · 984 阅读 · 0 评论