尤尔小屋

公众号：尤而小屋编辑：Peter作者：Peter大家好，我是Peter~本文介绍如何在PyTorch中创建Tensor，这是使用PyTorch进行深度学习的第一步，也是最为基础的一步。最后介绍PyTorch中常用的数据类型，以及如何进行设置和改变。

model.train() # 设置模型为训练模式for batch, (X,y) in enumerate(dataloader): # 遍历加载数据中的批次# 预测并计算误差# 基于反向传播进行参数优化loss.backward() # 反向传播计算梯度optimizer.step() # 更新模型参数optimizer.zero_grad() # 梯度清零if batch % 100 == 0: # 每100个批次打印一个损失值。