Update

diff --git a/beaver.py b/beaver.py
index a3a5615..5916215 100755 (executable)
--- a/beaver.py
+++ b/beaver.py
@@ -268,7 +268,6 @@ def oneshot(gpt, task):
                 policies.view(-1, 4, task.height, task.width),
             ).flatten(-2)
             scores = output
-            print(scores)
         else:
             raise ValueError(f"{args.oneshot_output=}")