learning curve pytorch