Browse Source

update model check

qingqu-git 1 year ago
parent
commit
45284e5ae2
3 changed files with 26 additions and 1 deletions
  1. 18 0
      deep_model_check.sh
  2. 7 0
      deep_model_start.sh
  3. 1 1
      rec_model_update.py

+ 18 - 0
deep_model_check.sh

@@ -0,0 +1,18 @@
+#!/bin/bash
+#mkdir -p /data/SortService
+file_info=/data/offline_dir/$1.tar
+filesize=$(stat -c%s ${file_info})
+#size=$(du -b ${file_info} | cut -f1)
+echo "The size of the file is $size bytes"
+maxsize=24368064001
+if [ $filesize -gt $maxsize ]
+then
+    echo "$filesize > $maxsize"
+    exit 0
+    #mv media.log media"`date +%Y-%m-%d_%H:%M:%S`".log
+else
+    echo "$filesize < $maxsize"
+    echi sdada
+    exit 1
+fi
+

+ 7 - 0
deep_model_start.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+mkdir -p /data/SortService
+tar xvf /data/offline_dir/$1.tar
+
+cp -r /data/offline_dir/models/$1/deepfm  /data/online_models/deepfm
+
+docker run -t --name deepfm -p 8501:8501 --mount type=bind,source=/data/online_models,target=/models -e MODEL_NAME=deepfm tensorflow/serving  > /data/SortService/server.log  &

+ 1 - 1
rec_model_update.py

@@ -116,7 +116,7 @@ def server_restart(slb_client, instance_id, ip_address, ins_name, dt):
         #image_id.sop()
         #print("image_id",image_id)
         server_start_sh = os.path.join(rec_model_config.start_sh['target_dir'], rec_model_config.start_sh['name'])
-        server_start_commend = f"sh {server_start_sh}"
+        server_start_commend = f"sh {server_start_sh} {dt}"
         utils.run_command(client=slb_client, instance_ids=instance_id_list, command=server_start_commend)
         logging.info(f"docker restart finished.")
         time.sleep(5)