2020年7月25日

pytorch 的LSTM batch_first=True 和 False的性能对比

摘要: pytorch 的LSTM batch_first=True 和 False的性能略有区别,不过区别不大。 下面这篇文章试验结论是batch_first= True要比batch_first = False更快。但是我自己跑结论却是相反,batch_first = False更快。 运行多次的结果: 阅读全文

posted @ 2020-07-25 12:34 那抹阳光1994 阅读(2544) 评论(0) 推荐(0) 编辑

pytorch中的parameter、variable和buffer

摘要: parameter 官网API 其可以将普通张量转变为模型参数的一部分。Parameters是Tensor的一个子类,当用于Module时具有非常特殊的属性,当其被赋予为模块的属性时,他们自动地添加到模块参数列表中,且将会出现在如parameters()迭代器中。如果赋予一个普通张量则没有这样的效果 阅读全文

posted @ 2020-07-25 11:34 那抹阳光1994 阅读(1140) 评论(0) 推荐(0) 编辑

导航