2019年7月30日
pytorch入门1——简单的网络搭建
代码如下： 上述代码完成了以下功能： 1、建立一个简单的网络，并给各个网络层的参数size进行赋值； 2、查看各个网络层参数量； 3、给网路一个随机的输入，查看网络输出； 4、查看网络每一层的额输出blob的大小； 这里需要注意的是，在进行第一个全连接层的定义时，self.fc1 = nn.Line Read More
posted @ 2019-07-30 20:31 rainsoul Views(1276) Comments(0) Diggs(0) Edit