Skip to content

Commit d6810ee

Browse files
author
bryan
committed
Fixed typos
1 parent dd5cb79 commit d6810ee

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ def load_data(is_train=True):
2121
Y: 3-D array of int. Has the shape of (# total games, 9, 9) (for train)
2222
or (batch_size, 9, 9) (for validation)
2323
'''
24-
X = np.load('data/sudoku.npz')['quizzes'].astype(np.float32)
25-
Y = np.load('data/sudoku.npz')['solutions']
24+
X = np.load('data/sudoku_quizzes.npy').astype(np.float32)
25+
Y = np.load('data/sudoku_solutions.npy')
2626

2727
X = np.expand_dims(X, -1)
2828

0 commit comments

Comments
 (0)