Add trainer flag step [ci skip] (#4147)
Add trainer flag step [ci skip] (#4147)

* Add trainer flag step

Add step to disable automatic optimization in the trainer
@justusschock

* Apply suggestions from code review

Co-authored-by: Nicki Skafte <6834dde87a397b36fec9709e4a16ddb6dba8b76b@gmail.com>

Co-authored-by: Jirka Borovec <b903e28b3fb12ead55c25f3a7816376be3bd063c@users.noreply.github.com>
Co-authored-by: Nicki Skafte <6834dde87a397b36fec9709e4a16ddb6dba8b76b@gmail.com>
williamFalcon/pytorch-lightning,williamFalcon/pytorch-lightning