Bingsu commited on
Commit
4c7fc08
1 Parent(s): 24c40e1

fix: onnx save path

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -64,7 +64,7 @@ def main():
64
 
65
  for path in paths:
66
  onnx = export(path)
67
- target_dir = save / Path(model).stem
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"