5x5专题

造车先做三蹦子-之二:自制数据集(5x5数据集)230102

#Jupyter Notebook231001import torchimport torch.nn as nnimport torch.optim as optim# 定义模型class Net(nn.Module):def __init__(self):super(Net, self).__init__()self.fc1 = nn.Linear(25, 50)self.fc2