Procházet zdrojové kódy

修改文件地址

zhy před 6 dny
rodič
revize
0f552f8394
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      tests/verify_tool_test.py

+ 1 - 1
tests/verify_tool_test.py

@@ -4,6 +4,6 @@
 from watermark_verify import verify_tool
 
 if __name__ == '__main__':
-    model_filename = "/mnt/e/DevTools/workspace/model_watermark_detect/tests/models/blackbox/alexnet/alexnet.onnx"
+    model_filename = "/home/ai/checkpoints/pytorch-alexnet-blackbox/om/model_99.om"
     verify_result = verify_tool.label_verification(model_filename)
     print(f"verify_result: {verify_result}")