fix: onnx save path
Browse files
models.py
CHANGED
@@ -64,7 +64,7 @@ def main():
|
|
64 |
|
65 |
for path in paths:
|
66 |
onnx = export(path)
|
67 |
-
target_dir = save / Path(
|
68 |
(target_dir / "1").mkdir(exist_ok=True, parents=True)
|
69 |
(target_dir / "config.pbtxt").touch()
|
70 |
target = target_dir / "1" / "model.onnx"
|
|
|
64 |
|
65 |
for path in paths:
|
66 |
onnx = export(path)
|
67 |
+
target_dir = save / Path(path).stem
|
68 |
(target_dir / "1").mkdir(exist_ok=True, parents=True)
|
69 |
(target_dir / "config.pbtxt").touch()
|
70 |
target = target_dir / "1" / "model.onnx"
|