Implement feature extraction and inference enhancements, including async processing and MongoDB integration

2026-01-30 14:43:30 +08:00
parent 3d78b88d47
commit 1bd4547b99
7 changed files with 166 additions and 3 deletions
--- a/inference.py
+++ b/inference.py
@@ -104,6 +104,13 @@ class InferenceEngine:
        """
        return self.inference_batch([json_path])

+# 配置参数
+backbone_dir = "Qwen3-1.7B"
+ckpt_path = "best_ckpt.pth"
+device = "cuda"
+
+engine = InferenceEngine(backbone_dir, ckpt_path, device)
+
 if __name__ == "__main__":
    # 配置参数
    backbone_dir = "Qwen3-1.7B"