torch.nn.Conv2d(in_channels, out_channels, kernel_size,
stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode=‘zeros’)
optimizer = optim.SGD(model.parameters(), lr = 0.01, momentum=0.9)
https://blog.csdn.net/qq_34690929/article/details/79932416?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522158808013819724839264889%2522%252C%2522scm%2522%253A%252220140713.130102334…%2522%257D&request_id=158808013819724839264889&biz_id=0&utm_source=distribute.pc_search_result.none-task-blog-2allfirst_rank_v2~rank_v25-2
保存训练的参数and调用
# 保存模型
t.save(net.state_dict(), 'net.pth')
# 加载已保存的模型
net2 = Net()
net2.load_state_dict(t.load('net.pth'))
#设置种子保证训练一致
torch.manual_seed(10)
torch.cuda.manual_seed(10)
torch.cuda.manual_seed_all(10)
np.random.seed(10)