X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=maze.py;fp=maze.py;h=cfdede39a59e0883b8c3781a5edd66381127ad35;hb=731aa9de1343e3e7bd5102b5e553d14893c0680a;hp=e377d2f0e6b2ab18684c93b6d85f7c6093cc67ea;hpb=27bb2d1ab23422f26b05f88b4e0573deeb075cd2;p=beaver.git

diff --git a/maze.py b/maze.py
index e377d2f..cfdede3 100755
--- 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)