pytorch 版本升级和接口修改

这里记录一下自己在运行pytorch 时由于版本修正引起的一些警告或错误：

 

1. 运行出现：

/usr/local/lib/python3.6/dist-packages/torch/nn/_reduction.py:43: UserWarning: size_average and reduce args will be deprecated, please use reduction='none' instead.
  warnings.warn(warning.format(ret))

解决办法：

criterion = nn.L1Loss(reduce=False, size_average=False)

改为：criterion = nn.L1Loss(reduction='none')

主要是loss函数里面的size_average, reduce这两个参数版本修改引起的，把这两个参数去掉改为reduction='none'即可