From 519b5419b30de82828a41b620e76d993a70423e9 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Fri, 7 Apr 2023 14:05:21 +0200 Subject: [PATCH] Update --- beaver.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/beaver.py b/beaver.py index 6ed9dd2..6e1eaf4 100755 --- a/beaver.py +++ b/beaver.py @@ -245,11 +245,14 @@ def oneshot(model, learning_rate_scheduler, task): -1, task.height, task.width ) mazes = mazes[:, : task.height * task.width].reshape(-1, task.height, task.width) - # targets = targets.reshape(-1, task.height, task.width) + paths = task.test_input[:32, task.height * task.width :].reshape( + -1, task.height, task.width + ) filename = f"oneshot.png" maze.save_image( os.path.join(args.result_dir, filename), mazes=mazes, + target_paths=paths, score_paths=proba_path, # score_truth=targets, ) -- 2.39.5