code style

Author: esythan

tools/onoff_diff/get_xbox_model.py

@@ -25,6 +25,7 @@
 
 #     fout.close()
 
+
 def get_xbox_model(infile, outfile):
     fout = open(outfile, 'w')
     out_list = ['0', '1', '0', '1']
@@ -51,11 +52,12 @@ def get_xbox_model(infile, outfile):
             fout.write('{}\t{}\n'.format(out_str1, out_str2))
     fout.close()
 
+
 if __name__ == '__main__':
     import argparse
     parser = argparse.ArgumentParser()
     parser.add_argument('-i', help='输入文件，cube结果', dest='infile', required=True)
     parser.add_argument(
         '-o', help='输出文件，xbox模型', dest='outfile', required=True)
     args = parser.parse_args()
-    get_xbox_model(args.infile, args.outfile)
+    get_xbox_model(args.infile, args.outfile)

tools/onoff_diff/onoff_diff.py

@@ -59,7 +59,7 @@ def get_data_from_log(online_log_file, origin_file):
                 res[insid] = {}
             res[insid][var_name] = (length, data)
     # print("online log:", res.keys())
-            
+
     return res
 
 
@@ -146,17 +146,20 @@ def onoff_max_diff(log_data, model_data, ins_id):
             if max_diff < diff:
                 max_diff = diff
         if max_diff > 2e-5:
-            print("ins_id:{}, var_name:{}: {}".format(ins_id, var_name, max_diff))
+            print("ins_id:{}, var_name:{}: {}".format(ins_id, var_name,
+                                                      max_diff))
 
 
 if __name__ == '__main__':
     import argparse
     parser = argparse.ArgumentParser()
-    parser.add_argument('-l', help='线上日志文件', dest='online_log_file', required=True)
+    parser.add_argument(
+        '-l', help='线上日志文件', dest='online_log_file', required=True)
     parser.add_argument(
         '-m', help='模型dump文件', dest='model_dump_file', required=True)
     parser.add_argument('-v', help='所有var文件', dest='var_file', required=True)
-    parser.add_argument('-o', help='offline原始数据文件', dest='origin_file', required=True)
+    parser.add_argument(
+        '-o', help='offline原始数据文件', dest='origin_file', required=True)
     args = parser.parse_args()
     log_data = get_data_from_log(args.online_log_file, args.origin_file)
     model_data = get_data_from_model(args.model_dump_file)

tools/onoff_diff/run.sh

@@ -39,7 +39,7 @@ fleetrun --server_num=1 --worker_num=1 ../../../tools/static_ps_offline_infer.py
 OFFLINE_DUMP_PATH=$MODEL_PATH/dump_offline_infer
 cat $OFFLINE_DUMP_PATH/part* > $TOOL_PATH/data/log.offline
 
-echo "online offline diff..."
+echo "online offline diff..." 
 cd $TOOL_PATH
 python3 onoff_diff.py -l ./data/log.online \
                       -m ./data/log.offline \

tools/static_ps_offline_infer.py

@@ -119,7 +119,7 @@ def run_offline_infer(self):
 
         self.exe.run(paddle.static.default_startup_program())
         fleet.init_worker()
-        
+
         init_model_path = config.get("runner.init_model_path")
         fleet.load_model(init_model_path, mode=0)