projects
/
picoclvr.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[picoclvr.git]
/
maze.py
diff --git
a/maze.py
b/maze.py
index
c2774dd
..
f6715f0
100755
(executable)
--- a/
maze.py
+++ b/
maze.py
@@
-324,9
+324,9
@@
def save_image(
if __name__ == "__main__":
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
if __name__ == "__main__":
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
- mazes, paths = create_maze_data(8)
+ mazes, paths
, policies
= create_maze_data(8)
mazes, paths = mazes.to(device), paths.to(device)
mazes, paths = mazes.to(device), paths.to(device)
- save_image("test.png", mazes
, paths,
paths)
+ save_image("test.png", mazes
=mazes, target_paths=paths, predicted_paths=
paths)
print(path_correctness(mazes, paths))
######################################################################
print(path_correctness(mazes, paths))
######################################################################