How to speed up? Which torch version is suggested #106
Description
I am testing Torch 2.0.1, 2.1 and 1.13.1
The speed changes but the results are also changing
For example 2.0.1 is slower but it ends quicker number of steps and produces better output
