move scheduler.step() to after validate

Aifred-Health / VulcanAI

A high level deep learning framework for quickly prototyping networks with added tools in data visualisation, model interpretability and performance metrics

https://www.aifredhealth.com

GNU Affero General Public License v3.0

17 stars 8 forks source link

move scheduler.step() to after validate #156

Closed nirtiac closed 5 years ago

nirtiac commented 5 years ago

Following breaking change https://github.com/pytorch/pytorch/pull/7889