新的信息

tools/optimize_model/main.py

@@ -0,0 +1,31 @@
+import paramiko,time,sys
+
+def main_optimize_model(pid):
+    try:
+        # 创建SSH对象
+        ssh = paramiko.SSHClient()
+        # 允许连接不在known_hosts文件中的主机
+        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+        print('start connect')
+        # 连接服务器
+        ssh.connect('connect.bjc1.seetacloud.com', username='root', password='QoQA8q3Ds2VB', port=33733)
+        print('connect success')
+
+        # 使用nohup执行命令，并将输出重定向到文件
+        command = f'nohup seg_python /data/code/optimize_model_xj/optimize_model.py -pid {pid} >{pid}.log 2>&1&'
+        stdin, stdout, stderr = ssh.exec_command(command)
+
+        # 打印命令是否启动成功
+        print("Command executed")
+    finally:
+        # 确保连接被关闭
+        if ssh:
+            time.sleep(1)
+            ssh.close()
+
+if __name__ == '__main__':
+
+    if len(sys.argv) == 2:
+        pid = sys.argv[1]
+        main_optimize_model(pid)
+