for batch_x, batch_y in train_loader:# 遍历每个batch。optimizer.zero_grad()# 重要！train_bat_loss.backward()# 自动计算梯度。with torch.no_grad():# 关闭梯度计算，节省内存和计算。# feature_data: 特征数据（比如表格的X部分）model.train()# 设置模型为训练模