Parcourir la source
`export.py` automatic `forward_export` (#6352)
* `export.py` automatic `forward_export`
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
modifyDataloader
Glenn Jocher
GitHub
il y a 2 ans
Parent
révision
750c42e43e
Aucune clé connue n'a été trouvée dans la base pour cette signature
ID de la clé GPG: 4AEE18F83AFDEB23
2 fichiers modifiés avec
3 ajouts et
2 suppressions
-
export.py
-
tutorial.ipynb
|
|
@@ -427,7 +427,8 @@ def run(data=ROOT / 'data/coco128.yaml', # 'dataset.yaml path' |
|
|
|
elif isinstance(m, Detect): |
|
|
|
m.inplace = inplace |
|
|
|
m.onnx_dynamic = dynamic |
|
|
|
# m.forward = m.forward_export # assign forward (optional) |
|
|
|
if hasattr(m, 'forward_export'): |
|
|
|
m.forward = m.forward_export # assign custom forward (optional) |
|
|
|
|
|
|
|
for _ in range(2): |
|
|
|
y = model(im) # dry runs |
|
|
@@ -1099,4 +1099,4 @@ |
|
|
|
"outputs": [] |
|
|
|
} |
|
|
|
] |
|
|
|
} |
|
|
|
} |