模型上线

train/train_service_sk_tf2_distibute.py

@@ -2,6 +2,7 @@ import tensorflow.compat.v1 as tf
 tf.logging.set_verbosity(tf.logging.INFO)
 import sys
 import time
+import shutil
 import os
 os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
 
@@ -272,6 +273,6 @@ filename = filenames[0]
 from datetime import date
 filename_ = date.fromtimestamp(int(filename)).strftime('%Y%m%d')
 
-os.rename(os.path.join(MODEL_BASE_DIR, filename), os.path.join(MODEL_BASE_DIR, filename_))
+shutil.move(os.path.join(MODEL_BASE_DIR, filename), os.path.join(MODEL_BASE_DIR, filename_))
 
 print("训练耗时: {}s".format(time.time() - start))
\ No newline at end of file