nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

[BUG] Scheduler.step() placement in default trainer #58

Closed alphinside closed 4 years ago

alphinside commented 4 years ago

Describe the bug Pytorch original scheduler have different application of scheduler step (some of them in each batch update, the others only called once per epoch ). However in default trainer, it always called per batch update

Expected behavior Every scheduler must apply the correct behavior of when to call the scheduler step()