Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
Summary
Add fuse function for RepConv
Motivation and Context
Fuse the multiple conv and bn layer, give me less latency and smaller model size
Dependencies
None
Type of change
For a new feature or function, please create an issue first to discuss it
with us before submitting a pull request.
Note: Please delete options that are not relevant.
Tests
An unit test function is added in
official/projects/yolo/modeling/layers/nn_blocks_test.RepConvTest, namedtest_fuse_and_unfuse_result.The test function checks the different of the two output values.
Test Configuration:
OS: ubuntu 16
Env:Python=3.10 TensorFlow=2.12.0
Checklist