Felix Marty
commited on
Commit
•
9f0bbf2
1
Parent(s):
78caf38
mention torch fx
Browse files
README.md
CHANGED
|
@@ -14,6 +14,8 @@ This model is a fork of [facebook/levit-256](https://huggingface.co/facebook/lev
|
|
| 14 |
|
| 15 |
and the optimized model is converted to the onnx format.
|
| 16 |
|
|
|
|
|
|
|
| 17 |
## How to use
|
| 18 |
|
| 19 |
```python
|
|
|
|
| 14 |
|
| 15 |
and the optimized model is converted to the onnx format.
|
| 16 |
|
| 17 |
+
The fusion of layers leverages torch.fx, using the transformations `FuseBatchNorm2dInConv2d` and `FuseBatchNorm1dInLinear` soon to be available to use out-of-the-box with 🤗 Optimum, check it out: https://huggingface.co/docs/optimum/main/en/fx/optimization#the-transformation-guide .
|
| 18 |
+
|
| 19 |
## How to use
|
| 20 |
|
| 21 |
```python
|