本文共 627 字,大约阅读时间需要 2 分钟。
mobilefacenet:
prelu 平均106ms
relu 平均94ms,快12ms,快10%
relu6 平均97ms,收敛慢
leakyrelu,平均106ms,收敛更慢。
inplace=True会x把输入前的删掉了,都变成了输入后的,
没有inplace,输入还是输入,输出是新的了。
import torchimport torch.nn as nn#inplace为True,将会改变输入的数据 ,否则不会改变原输入,只会产生新的输出m = nn.LeakyReLU(inplace=True)input = torch.randn(7)print("输入处理前图片:")print(input)output = m(input)print("ReLU输出:")print(output)print("输入处理后图片:")print(input)
import torchimport torch.nn.functional as Fconf_mask = torch.FloatTensor([0.0, 1.0, 0.0, 1.0, 1.0])conf_data = torch.FloatTensor([-20.1, -0.9, 0.0, 0.2,0.5,0.99, 2.2])import torchimport torch.nn as nn#inplace为True,将会改变输入的数据 ,否则不会改变原输入,只会产生新的输出
转载地址:http://tpak.baihongyu.com/