I would recommend to compare the number of parameters between both models and make sure they are equal (if not already done).



 mohaimen:

My understanding is that torch should be around 2x faster.


Why should this be the case?