From 13e4bbafc59edd81528ddf8320b58052daec50b8 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Tue, 21 Mar 2023 07:01:11 +0100
Subject: [PATCH] Update

---
 README.txt | 11 +++--------
 beaver.py  |  2 +-
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/README.txt b/README.txt
index 31710e2..dc13a4f 100644
--- a/README.txt
+++ b/README.txt
@@ -1,9 +1,4 @@
-To run the experiments:
+To train the shortest-path solving GPT, and train the one-shot MLP
+read-out:
 
-  ./beaver.py
-
-to train the shortest-path solving GPT, and
-
-  ./beaver.py --overwrite --one_shot --nb_epochs=10
-
-to train the one-shot MLP read-out.
+  ./beaver.py --oneshot
diff --git a/beaver.py b/beaver.py
index f5bd924..bdc12aa 100755
--- a/beaver.py
+++ b/beaver.py
@@ -86,7 +86,7 @@ parser.add_argument("--oneshot", action="store_true", default=False)
 
 parser.add_argument("--oneshot_input", type=str, default="head")
 
-parser.add_argument("--oneshot_output", type=str, default="policy")
+parser.add_argument("--oneshot_output", type=str, default="trace")
 
 ######################################################################
 
-- 
2.39.5