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