Weights at epoch 64 not found about dcrnn_pytorch HOT 12 CLOSED

Setting train/epoch to 0 in data/model/dcrnn_test_config.yaml solves the problem for me.

from dcrnn_pytorch.

@chnsh actually I ended up doing the change in data/model/pretrained/METR-LA/config.yaml and that did it

from dcrnn_pytorch.

@zhenhuascut @victorsoda
"'models/epo64.tar' " is not provided in this branch........

from dcrnn_pytorch.

Thanks, I have solved it.

from dcrnn_pytorch.

Still don't see epo64.tar ... It would be great if you could upload it. Many thanks!

from dcrnn_pytorch.

Still don't see too... Would you upload it then I could reproduce your work? Thanks a lot.

from dcrnn_pytorch.

I also got the same trouble, but at 51 epoch. Can you help me solve it? Thanks a lot.

from dcrnn_pytorch.

Setting train/epoch to 0 in data/model/dcrnn_test_config.yaml solves the problem.

Thanks! I tried successfully.

from dcrnn_pytorch.

Hi everyone, I am sorry I have been unable to reply - @htn274 is correct - the epoch is intended to be the checkpoint from where to resume and I have since then shut down the server (I am a poor student 🥼) and haven't been able to retrieve the weights at epoch 64 - if you just set it to 0 and train, you will be able to reproduce the results

from dcrnn_pytorch.

I don't believe setting train/epoch to 0 is a solution for this problem. As you train the model by the code, the trained models are saved as models/epo0.tar, epo1.tar,... etc. Therefore if you run "run_demo" with train/epoch = 0, it means you run demo with the trained model only with the first epoch. So I ask you to add the best model (epoXX.tar) at the models/ for METR-LA, and PEMS-BAY then we can test with these models

from dcrnn_pytorch.

@semink that is incorrect - the model only tries to load existing weights if epoch > 0, so by setting epoch=0 will do the job as @baosws helpfully pointed out.

I am closing this issue for now - the solution is to train it and once it has trained, set the correct epoch number in config.yml and that should work

from dcrnn_pytorch.

@chnsh this didn't work for me. I still get the same error

from dcrnn_pytorch.