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