To train the model, simply open a Terminal, cd to tensorflow/models/tutorials/rnn/quickdraw, then run the following script:
python train_model.py \ --training_data=rnn_tutorial_data/training.tfrecord-?????-of-????? \ --eval_data=rnn_tutorial_data/eval.tfrecord-?????-of-????? \ --model_dir quickdraw_model/ \ --classes_file=rnn_tutorial_data/training.tfrecord.classes
By default, the training steps are 100k, and it takes about 6 hours on our GTX 1070 GPU to finish the training. After the training is completed, you'll see a familiar file listing in the model directory (omitting the other four sets of model.ckpt* files):
ls -lt quickdraw_model/-rw-rw-r-- 1 jeff jeff 164419871 Feb 12 05:56 events.out.tfevents.1518422507.AiLabby ...
