-
I try to follow the example to prune the transformer model, but the layer-norm always gets a size mismatch error.
-
## 🐛 Bug
Hi, I was working on iteratively pruning with global_pruning method (training --> prune --> training --> prune). However, I found it costs GPU memory after pruning.
With a very big mode…
-
Hi, thank you for publishing this amazing work about the structural pruning on diffusion models. I wondered if you are also publishing the code for diff-pruning and training on LDM. The current ldm-pr…
-
**Describe the bug**
prune_low_magnitude cannot reduce the size of tflite model. I run the pruning with keras example (https://github.com/tensorflow/model-optimization/blob/master/tensorflow_model_…
-
Pruning a model with GLU results in an error when finding importance. GLU does not have any params but halves the input (in the given dimension). This is not accounted for during tracing, assigning in…
-
@VainF Thanks for amazing repo.
I tried to run inference for yolov7 pruned model on one image, **but pruned model did not detect anything** (output image without bboxes) https://github.com/VainF/Torc…
-
Hi
when I use 'tiny_yolo_v3' pretrained model in train.py, I got the following error.could you help me to solve that?
Code :
!python train.py --model_type=tiny_yolo3_darknet --anchors_path=configs/…
-
Add a `doctr.models.utils` module to compress existing models and improve their latency / memory load for inference purposes on CPU. Some interesting leads to investigate:
- [x] FP conversion (#10)
…
-
# Description
HistFactory includes a threshold setting for the size of statistical uncertainties related to predicted distributions, for example
```
```
taken from [here](https://github.com/ale…
-
**System information**
- TensorFlow version (you are using): 2.5.0
- Are you willing to contribute it (Yes/No): Yes
**Motivation**
There is pruning result of GNMT EN-DE and GNMT DE-EN on htt…