Did you benchmark after the first run? Pytorch uses optimization on the first forward pass so it usually takes longer by design.
Did you benchmark after the first run? Pytorch uses optimization on the first forward pass so it usually takes longer by design.