zhy 6 дней назад
Родитель
Сommit
0f552f8394
1 измененных файлов с 1 добавлено и 1 удалено
  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}")