pytorch

201 阅读1分钟

1.损失函数

(1)nn.MSELoss()

有三个可选参数:reduce、size_average、reduction

  • 如果 reduce = False,那么 size_average 参数失效,直接返回向量形式的 loss

  • 如果 reduce = True,那么 loss 返回的是标量

    a)如果 size_average = True,返回 loss.mean(),即loss的平均值

    b)如果 size_average = False,返回 loss.sum(),loss的和

  • reduction = ‘none’,直接返回向量形式的 loss

  • reduction = ‘sum’,返回loss之和

  • reduction = ''elementwise_mean,返回loss的平均值

  • reduction = ''mean,返回loss的平均值

2. _ _ init _ _ ; forward ; _ _ call _ _

3. epoch batch.size iteration