Commit
•
e789019
1
Parent(s):
fa52fdd
Update tools/demo.py
Browse files- tools/demo.py +1 -0
tools/demo.py
CHANGED
|
@@ -116,6 +116,7 @@ def export_model(runner,
|
|
| 116 |
# dry run
|
| 117 |
deploy_model(fake_input)
|
| 118 |
|
|
|
|
| 119 |
save_onnx_path = os.path.join(
|
| 120 |
args.work_dir,
|
| 121 |
os.path.basename(args.checkpoint).replace('pth', 'onnx'))
|
|
|
|
| 116 |
# dry run
|
| 117 |
deploy_model(fake_input)
|
| 118 |
|
| 119 |
+
os.makedirs(args.work_dir, exist_ok=True)
|
| 120 |
save_onnx_path = os.path.join(
|
| 121 |
args.work_dir,
|
| 122 |
os.path.basename(args.checkpoint).replace('pth', 'onnx'))
|