最近公司使用算法要用pytorch,所以本人暂时放弃使用tensorflow,为了练手pytorch,本人首先使用pytorch将tensorflow版本的mnist转换成pytorch版本,tensorflow原版本如下所示:
1 | from tensorflow.examples.tutorials.mnist import input_data |
改写成pytorch版本后如下所示:
1 | import torch.nn as nn |
几点感受:
1.pytorch不需要定义静态图,可以直接上手使用,网络定义更加方便快捷。
2.许多功能定义封装非常完善,不需要自己手写。
3.变量的定义和训练使用非常方便。
参考:
1.https://blog.csdn.net/victoriaw/article/details/72354307
2.https://blog.csdn.net/caichao08/article/details/78997033
3.https://blog.csdn.net/sparta_117/article/details/66965760