Update

[beaver.git] / maze.py

diff --git a/maze.py b/maze.py
index e377d2f..cfdede3 100755 (executable)
--- a/maze.py
+++ b/maze.py
@@ -241,7 +241,6 @@ def save_image(name, mazes, target_paths, predicted_paths=None, path_correct=Non
 ######################################################################
 
 if __name__ == "__main__":
-
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     mazes, paths = create_maze_data(8)
     mazes, paths = mazes.to(device), paths.to(device)