You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Signle decode cost 137 ms per image
Batch decode cost 0.64 ms per image
I guess the reason is the torchnvjpeg.Decoder parameter different, so I set the param
without torch.cuda.current_stream(device_id), time is 137 ms
with torch.cuda.current_stream(device_id), time is 69 ms
So I suggest use below to initialize. decoder = torchnvjpeg.Decoder(0, 0, True, device_id, batch_size, max_cpu_threads, max_image_size, torch.cuda.current_stream(device_id))
The text was updated successfully, but these errors were encountered:
Signle decode cost 137 ms per image
Batch decode cost 0.64 ms per image
I guess the reason is the torchnvjpeg.Decoder parameter different, so I set the param
without torch.cuda.current_stream(device_id), time is 137 ms
with torch.cuda.current_stream(device_id), time is 69 ms
So I suggest use below to initialize.
decoder = torchnvjpeg.Decoder(0, 0, True, device_id, batch_size, max_cpu_threads, max_image_size, torch.cuda.current_stream(device_id))
The text was updated successfully, but these errors were encountered: