evit icon indicating copy to clipboard operation
evit copied to clipboard

Bug in speed_test.py

Open caddyless opened this issue 2 years ago • 1 comments

I found torch.no_grad() is missed in the speed_test.py, which significantly slows down the throughput. Is this intentional or a mistake?

caddyless avatar Sep 23 '22 08:09 caddyless

Hi @caddyless,

Thanks for your comment. Actually, I didn't observe notable differences in the results of the throughput testing with/without torch.no_grad(). The differences were all within 1%, so there was no need to add torch.no_grad(). Did you find a large difference in the throughput testing with/without torch.no_grad()?

youweiliang avatar Sep 24 '22 06:09 youweiliang