nn.ReLU() 和 nn.ReLU(inplace=True)区别

inplace=True
计算结果不会有影响。利用in-place计算可以节省内(显)存,同时还可以省去反复申请和释放内存的时间。但是会对原变量覆盖,只要不带来错误就用。

猜你喜欢

转载自blog.csdn.net/qq_35608277/article/details/85881897