Update multi_dcu_train.py

multi_dcu_train.py

@@ -149,6 +149,13 @@ if __name__=='__main__':
     beam_size = 4
     no_repeat_ngram_size = 2
 
+    folder_path = "/saves/train_dtk_weights"
+
+    # 检查文件夹是否存在
+    if not os.path.exists(folder_path):
+        # 如果不存在，则创建文件夹
+        os.makedirs(folder_path)
+
     train_data = LCSTS('/umt5/data/lcsts_tsv/data1.tsv')
     valid_data = LCSTS('/umt5/data/lcsts_tsv/data2.tsv')