[Minor] Simplify training procedure - flatten function calls (#1642)

* flatten training function calls

* fix order

* remove commented _train()

* further flatten and reset model after LR finder

* undo model reset

* finish trying to reset model