Lr scheduler resets when resuming model

chnsh / DCRNN_PyTorch

Diffusion Convolutional Recurrent Neural Network Implementation in PyTorch

MIT License

456 stars 115 forks source link

Lr scheduler resets when resuming model #5

Open Noahprog opened 4 years ago

Noahprog commented 4 years ago

Hi, many thanks for converting the tensorflow implementation of DCRNN to pytorch, it helped me a lot.

I noticed that when resuming a model, the learning-rate scheduling is being reset. Any idea why this happens? I uploaded a picture of a test case below. If you need any further details, let me know.

chnsh commented 4 years ago

That's a good find - can you submit a PR to fix it? @Noahprog