class CONCAT_CNN(nn.Module):
def __init__(self):
super(CONCAT_CNN, self).__init__()
self.conv1_1 = nn.Conv2d(1, 64, kernel_size=3, stride=1, padding=1)
def forward(self, x):
conv1_1 = F.relu(self.conv1_1
学习笔记2:Pytorch 神经网络初始化权重与偏置
最新推荐文章于 2024-01-12 15:20:13 发布