To train the shortest-path solving GPT, and train the one-shot MLP
read-out:

  ./beaver.py --oneshot

Same, lighter settings (~95% test success instead of ~99%):

  ./beaver.py --nb_train_samples=25000 --nb_test_samples=10000 --nb_epochs=10 --oneshot