摘要:
Utils 从torch.nn.utils模块 clip_grad_norm_ 裁剪参数迭代的梯度范数。 clip_grad_value_ 在指定值处剪辑可迭代参数的梯度。 parameters_to_vector 将参数转换为一个向量 vector_to_parameters 将一个向量转换为参数 阅读全文
摘要:
卷积层 nn.Conv1d 对由多个输入平面组成的输入信号应用一维卷积。 nn.Conv2d 在由多个输入平面组成的输入信号上应用 2D 卷积。 nn.Conv3d 对由多个输入平面组成的输入信号应用 3D 卷积。 nn.ConvTranspose1d 在由多个输入平面组成的输入图像上应用一维转置卷 阅读全文
摘要:
损失函数 nn.L1Loss 创建一个衡量输入中每个元素之间的平均绝对误差 (MAE) 的标准XX和目标是的是的. nn.MSELoss 创建一个标准,用于测量输入中每个元素之间的均方误差(平方 L2 范数)XX和目标是的是的. nn.CrossEntropyLoss 该标准计算输入和目标之间的交叉 阅读全文