Browse Source

feat: 修改设备参数为通用GPU配置以提高兼容性,
一页多个表格有问题,只输出了最后一个表格

zhch158_admin 1 tháng trước cách đây
mục cha
commit
0cb4e1e389
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      zhch/ppstructurev3_single_process.py

+ 1 - 1
zhch/ppstructurev3_single_process.py

@@ -352,7 +352,7 @@ if __name__ == "__main__":
             # "output_dir": "./OmniDocBench_PPStructureV3_Results",
             # "collect_results": f"./OmniDocBench_PPStructureV3_Results/processed_files_{time.strftime('%Y%m%d_%H%M%S')}.csv",
             "pipeline": "./my_config/PP-StructureV3.yaml",
-            "device": "gpu:3",
+            "device": "gpu",
         }
         
         # 构造参数