模型上线

ab1e8bc0 · 宋柯 · b8da2eaf · ab1e8bc0
Commit ab1e8bc0 authored 3 years ago by 宋柯
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

train_service_sk_tf2_distibute.py train/train_service_sk_tf2_distibute.py +2 -2

No files found.
--- a/train/train_service_sk_tf2_distibute.py
+++ b/train/train_service_sk_tf2_distibute.py
@@ -9,8 +9,8 @@ start = time.time()

 DATA_DIR = '/data/files/wideAndDeep/trainData/'
 CHECKPOINTS_DIR = '/data/files/wideAndDeep/checkpoints/'
-if os.path.exists(CHECKPOINTS_DIR):
-    os.rmdir(CHECKPOINTS_DIR)
+# if os.path.exists(CHECKPOINTS_DIR):
+#     os.rmdir(CHECKPOINTS_DIR)

 MODEL_BASE_DIR = '/data/files/wideAndDeep/tensorServing_models/service'