From 15d0ad06d29c3e8042bdf64b78a01572e9b7ddc5 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Mon, 20 Mar 2023 22:42:19 +0100 Subject: [PATCH] Update --- beaver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/beaver.py b/beaver.py index f62c749..a3a5615 100755 --- a/beaver.py +++ b/beaver.py @@ -229,7 +229,6 @@ def oneshot(gpt, task): # s = maze.stationary_densities( # exit(0) #### - masks = mazes == maze.v_empty output_gpt = gpt(mygpt.BracketedSequence(mazes), mode=args.oneshot_input).x output = model(output_gpt) @@ -268,7 +267,8 @@ def oneshot(gpt, task): mazes.view(-1, task.height, task.width), policies.view(-1, 4, task.height, task.width), ).flatten(-2) - scores = output.flatten(-2) + scores = output + print(scores) else: raise ValueError(f"{args.oneshot_output=}") -- 2.20.1