1. nn.ReLU() 和 nn.ReLU(inplace=True)区别
inplace=True
计算结果不会有影响。利用in-place计算可以节省内(显)存,同时还可以省去反复申请和释放内存的时间。但是会对原变量覆盖,只要不带来错误就用。
2. 详解pytorch中nn模块的BatchNorm2d()函数
https://blog.csdn.net/bigFatCat_Tom/article/details/91619977
1. nn.ReLU() 和 nn.ReLU(inplace=True)区别
inplace=True
计算结果不会有影响。利用in-place计算可以节省内(显)存,同时还可以省去反复申请和释放内存的时间。但是会对原变量覆盖,只要不带来错误就用。
2. 详解pytorch中nn模块的BatchNorm2d()函数
https://blog.csdn.net/bigFatCat_Tom/article/details/91619977