在文档中解释是:

参数: inplace-选择是否进行覆盖运算

意思是是否将得到的值计算得到的值覆盖之前的值,比如:

x = x +1

即对原值进行操作,然后将得到的值又直接复制到该值中

而不是覆盖运算的例子如:

y = x + 1
x = y

这样就需要花费内存去多存储一个变量y

 

所以

nn.Conv2d(3, 64, kernel_size=11, stride=4, padding=2),
nn.ReLU(inplace=True)

的意思就是对从上层网络Conv2d中传递下来的tensor直接进行修改,这样能够节省运算内存,不用多存储其他变量

相关文章:

  • 2022-02-25
  • 2021-08-22
  • 2022-12-23
  • 2021-06-10
  • 2022-12-23
  • 2022-03-01
  • 2021-11-20
  • 2022-01-05
猜你喜欢
  • 2021-11-25
  • 2022-12-23
  • 2021-09-30
  • 2021-07-09
  • 2022-12-23
  • 2021-10-07
  • 2022-12-23
相关资源
相似解决方案