目录
原版测试代码:
1060显卡,224*224,平均时间10ms以内。
import time
import torch
from unireplknet import unireplknet_a
if __name__ == '__main__':
net = unireplknet_a()
net.cuda()
net.eval()
for i in range(20):
x = torch.randn(2, 3, 224, 224).cuda()
start=time.time()
out=net(x)
print(out.size(),time.time()-start)