天天看點

pytorch h-sigmoid

import torch x = torch.rand(30, 3, 416, 416).cuda() start=time.time() torch.sigmoid(x) print(f'time,{time.time()-start}')

import torch

x = torch.rand(30, 3, 416, 416).cuda()
aaaaa=torch.nn.ReLU6()
start=time.time()
torch.sigmoid(x)
# x=aaaaa(x+3)/6

print(f'time,{time.time()-start}')