pytorch构建网络模型的4种方法
利用pytorch来构建网络模型有很多种方法,以下简单列出其中的四种。 假设构建一个网络模型如下: 卷积层–》Relu层–》池化层–》全连接层–》Relu层–》全连接层 首先导入几种方法用到的包: import torch import torch.nn.functional as F from collections import OrderedDict 第一种方法 # Method 1 ----------------------------------------- class Net1(torch.nn.Module): def __init__(self):