From 13e4bbafc59edd81528ddf8320b58052daec50b8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= 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