Ok I will try that. Can I conclude that for pytorch, that’s the fastest I can get? 5x slower?
It is quite pointless to compare with a slower version of another framework. I only care about the fastest possible speed.