1. 基本步骤

for epoch in range(num_epochs):for _X, _y in dataloader:# 将数据转移到GPU_X, _y = _X.to(device), _y.to(device)# 前向传播计算损失loss = criterion(model(_X).reshape(_y.shape), _y)# 清空优化器梯度缓存optimizer.zero_grad()# 误差反向传播loss.backward()# 优化器更新参数optimizer.step()# 计算参数更新后模型的损失loss = criterion(model(X).reshape(y.shape), y)