Update quick start.
@@ -41,14 +41,14 @@ Now let's try to train a model on it (even on your laptop, this should train fas
.. code-block:: bash
# train MemNN using batch size 8 and 4 threads for 5 epochs
- python examples/train_model.py -t babi:task10k:1 -mf /tmp/babi_memnn -bs 8 -nt 4 -eps 5 -m memnn
+ python examples/train_model.py -t babi:task10k:1 -mf /tmp/babi_memnn -bs 8 -nt 4 -eps 5 -m memnn --no-cuda
Let's print some of its predictions to make sure it's working.
.. code-block:: bash
# display predictions for model save at specified file on bAbI task 1
- python examples/display_model.py -t babi:task10k:1 -mf /tmp/babi_memnn
+ python examples/display_model.py -t babi:task10k:1 -mf /tmp/babi_memnn -ecands vocab
The "eval_labels" and "MemNN" lines should (usually) match!
