68

app/data/shakespeare_char/prepare.py

@@ -51,7 +51,15 @@ val_ids = np.array(val_ids, dtype=np.uint16)
 
 file_path_train = os.path.join(os.path.dirname(__file__), 'train.bin')
 file_path_val = os.path.join(os.path.dirname(__file__), 'val.bin')
+
+if not os.path.exists(file_path_train):
+    with open(file_path_train, 'wb') as file:
+
+if not os.path.exists(file_path_val):
+    with open(file_path_val, 'wb') as file:
+
 permissions = 0o755
+
 os.chmod(file_path_train, permissions)
 os.chmod(file_path_val, permissions)