Selaa lähdekoodia

refactor: 注释掉默认配置,优化代码可读性

zhch158_admin 2 kuukautta sitten
vanhempi
commit
ac06e6b3d5
1 muutettua tiedostoa jossa 8 lisäystä ja 8 poistoa
  1. 8 8
      zhch/ppstructurev3_single_process.py

+ 8 - 8
zhch/ppstructurev3_single_process.py

@@ -314,14 +314,14 @@ if __name__ == "__main__":
             "collect_results": "./OmniDocBench_PPStructureV3_Results/processed_files.csv",
         }
         
-        default_config = {
-            "input_csv": "./OmniDocBench_PPStructureV3_Results/processed_files.csv",
-            "output_dir": "./OmniDocBench_PPStructureV3_Results",
-            "pipeline": "PP-StructureV3",
-            "device": "gpu:0",
-            "batch_size": 2,
-            "collect_results": f"./OmniDocBench_PPStructureV3_Results/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
-        }
+        # default_config = {
+        #     "input_csv": "./OmniDocBench_PPStructureV3_Results/processed_files.csv",
+        #     "output_dir": "./OmniDocBench_PPStructureV3_Results",
+        #     "pipeline": "PP-StructureV3",
+        #     "device": "gpu:0",
+        #     "batch_size": 2,
+        #     "collect_results": f"./OmniDocBench_PPStructureV3_Results/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
+        # }
         # 构造参数
         sys.argv = [sys.argv[0]]
         for key, value in default_config.items():