部分训练代码(损失函数及之前的相关代码示意):
def train_cnn():
model = CNN_model()
train_set = Data.DataLoader(dataset=loadDataset('./dataset/train_dataset.npy', './dataset/train_labels.npy'), batch_size=128, shuffle=True)
optimizer = torch.optim.Adam(model.parameters(), lr=LR