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"
|