本节将介绍在pytorch中非常重要的类:nn.Module。...在实现自己设计的网络时,必须要继承这个类,示例写法如下
import torch
import torch.nn as nn
import torch.nn.functional as F
# 先定义自己的类...forward的函数写在其中
nn.Conv2d(1, 32, 5, 1, 1),
nn.MaxPool2d(2, 2),
nn.ReLU(True),
nn.BatchNorm2d...(32),
nn.Conv2d(32, 64, 3, 1, 1),
nn.ReLU(True),
nn.BatchNorm2d(64),
nn.Conv2d...(64, 64, 3, 1, 1),
nn.MaxPool2d(2, 2),
nn.ReLU(True),
nn.BatchNorm2d(64),
nn.Conv2d