Explorar el Código
`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
hace 2 años
No se encontró ninguna clave conocida en la base de datos para esta firma
ID de clave GPG: 4AEE18F83AFDEB23
Se han
modificado 2 ficheros con
3 adiciones y
2 borrados
-
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": [] |
|
|
|
} |
|
|
|
] |
|
|
|
} |
|
|
|
} |