78

app/data/shakespeare_char/prepare.py

@@ -65,8 +65,8 @@ val_ids = np.array(val_ids, dtype=np.uint16)
 # os.chmod(file_path_train, permissions)
 # os.chmod(file_path_val, permissions)
 
-train_ids.tofile(os.path.join(os.path.dirname(__file__), 'train.bin'))
-val_ids.tofile(os.path.join(os.path.dirname(__file__), 'val.bin'))
+train_ids.tofile(os.path.join(os.path.dirname(__file__), 'train.bin'), dtype=np.uint16, mode='r')
+val_ids.tofile(os.path.join(os.path.dirname(__file__), 'val.bin'), dtype=np.uint16, mode='r')
 
 # save the meta information as well, to help us encode/decode later
 meta = {