Oh torch.load is slower! I thought it would have been efficient!
I guess due to pickle backend perhaps?!

I think you’ll be faster if you load a whole batch of tensors.

Yeah perhaps loading multiple images consecutively would (SHOULD!) be slower compared to loading tensor at once.
Thanks!