X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=beaver.py;h=6e1eaf4137f5a3bc184f4bd8fd529adbd1791745;hb=519b5419b30de82828a41b620e76d993a70423e9;hp=6ed9dd23a384984374e8feff2ba489d103150606;hpb=92a2935401f6fe21efe19ac3c476521665a242a5;p=beaver.git

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,
     )