nncf icon indicating copy to clipboard operation
nncf copied to clipboard

BootstrapNAS: Export best subnet to onnx

Open jpablomch opened this issue 2 years ago • 3 comments

Changes

Add calls in examples to export the resulting best subnet to ONNX

Reason for changes

Example of exporting to ONNX requested by user.

Related tickets

N/A

Tests

Test for exporting to ONNX are already included in test suite.

jpablomch avatar Aug 09 '22 22:08 jpablomch

@ljaljushkin Please take a look. Thanks!

jpablomch avatar Aug 09 '22 22:08 jpablomch

This PR can't be merged without passing all pre-commit checks.

There's issue with code style, please refer to pylint check log.

Also, one e2e test for efficient (nas.test_sanity_sample.test_e2e_supernet_training[efficient_net_b1_cifar10_nas.json_classification_nas_mock_32x32]) is failing with error: RuntimeError: ONNX export failed: Couldn't export Python operator SwishImplementation.

Seems like there's an option to use export-friendly version of this model: https://github.com/lukemelas/EfficientNet-PyTorch/issues/91#issuecomment-542994572 Please use that version in the test.

ljaljushkin avatar Aug 17 '22 16:08 ljaljushkin

Can one of the admins verify this patch?

openvino-nncf-ci avatar Oct 10 '22 17:10 openvino-nncf-ci

run build

aiarkinx avatar Oct 27 '22 12:10 aiarkinx

run build

jpablomch avatar Oct 27 '22 21:10 jpablomch

run build

jpablomch avatar Nov 10 '22 22:11 jpablomch

run build

jpablomch avatar Nov 28 '22 05:11 jpablomch

run build

jpablomch avatar Nov 29 '22 03:11 jpablomch