search

ml5js / ml5-data-and-models

Data sets and pre-trained models for ml5.js
https://ml5js.org/docs/data-overview
MIT License
125 stars 98 forks source link

LSTM train.py not loading from checkpoint #27

Open karooolis opened 6 years ago

karooolis commented 6 years ago

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 commented 6 years ago

Was the checkpoint generated from the same script?

karooolis commented 6 years ago

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 commented 6 years ago

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