search

weinman / cnn_lstm_ctc_ocr

Tensorflow-based CNN+LSTM trained with CTC-loss for OCR
GNU General Public License v3.0
498 stars 170 forks source link

FineTune ! #58

Closed imohammadhossein closed 5 years ago

imohammadhossein commented 5 years ago

hey thanks for your codes again :) the question is that is it posiible to finetune the trained model fom your networks with your codes ?

weinman commented 5 years ago

If you're asking whether I have pretrained models, the answer is no (See #40 and #45). If the question is how do you train only some of the weights, then you train using the --tune_from flag to indicate the initial model weights and --tune_scope to restrict the categories of weights that you want to adjust. For example, all the LSTM weights have the variable scope "rnn" (e.g., see this line in model.py.

imohammadhossein commented 5 years ago

how to use the flag in command ? can you make an example ?

weinman commented 5 years ago

@imohamadhoseins See the newfangled FAQ.