jupyter日常测性能

发表于 2021-04-15 阅读次数：
本文字数： 161 阅读时长 ≈ 1 分钟

在jupyter中日常测试不同表达方式的执行效率，进而做取舍时，直接新开一行以

%%timeit

开头即可

1 2	way1=LossFunc() way1=way1.cuda()

1 2	%%timeit tmp=way1(torch.rand(10000,1),torch.rand(10000,1))