To train the shortest-path solving GPT, and train the one-shot MLP read-out: ./beaver.py --oneshot