Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

LSTM train.py not loading from checkpoint #27

Open
karooolis opened this issue Jun 18, 2018 · 3 comments
Open

LSTM train.py not loading from checkpoint #27

karooolis opened this issue Jun 18, 2018 · 3 comments

Comments

@karooolis
Copy link

karooolis commented Jun 18, 2018

Hi, I try to start training a model from a checkpoint. I use init_from option but somehow the model still starts training from 0. What could be the problem here?

The sample command looks like this - python train.py --data_dir=/home/ubuntu/ml5-data-and-training/datasets/text/famous_quotes --rnn_size 256 --num_layers 2 --seq_length 64 --batch_size 32 --output_keep_prob 0.25 --init_from=/home/ubuntu/ml5-data-and-training/training/lstm/checkpoints/famous_quotes.

@cvalenzuela
Copy link
Member

Was the checkpoint generated from the same script?

@karooolis
Copy link
Author

I must have not pressed the comment button when I commented last time. Yes, the checkpoint was generated with exactly the same script. Have you heard of anyone else having similar issues?

@cvalenzuela
Copy link
Member

Not sure what could be the issue. A new repo with some fixes was recently published here: https://github.com/ml5js/training-lstm, you could try that insted

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants