-
is there any example about pruning model on multi GPUs? I had tried prune mode with multi GPU, but it failed(forget to collect log), then I prune model with one GPU and save model to disk, after that,…
-
### Validations
- [X] I believe this is a way to improve. I'll try to join the [Continue Discord](https://discord.gg/NWtdYexhMs) for questions
- [X] I'm not able to find an [open issue](https://gi…
-
When I use this YOLOv8 pruning routine, I cannot carry out deepcopy, I use OBB model, I suspect it is caused by hook operation, but I do not know how to solve it.
BUG:Only Tensors created explicitly …
-
Hi, I want to ask that what is the strategy at inference stage ? The method utilizes gumbel softmax to generate preserve token mask at training stage, but how to generate preserve token mask at infere…
-
![image](https://github.com/TheLastBen/fast-stable-diffusion/assets/127377346/bc47dbe6-d995-4135-bd11-54649ab279bc)
Hello, I need to merge checkpoints, but whenever I try to do this, this error happe…
-
I've been working for a couple of months with VAI Optimizer and after a careful reading of both Vitis AI Optimizer User Guide (UG1333) (v3.0) and Vitis AI User Guide (UG1414) (v3.5) I have some doubts…
-
Can we utilise train_pruning.py to prune other pytorch models from Ross Wightman repository? For eg. SEResNeXT26t or CoATNet
-
Hello, I would like to ask how I can use your pruning method to prune the swin-t transformer-tiny model to get a model after pruning, I see that your search architecture part is used to search how to …
bo102 updated
5 months ago
-
I am trying to apply filter pruning with LeGR to yolov8 model. It was able to run the pruning itself, but I have few questions about the implementation and result.
1. In train_step during p…
-
I just started using NNCF and I used NNCF to do filter pruning on Yolov5s. When I trained the model according to the steps in `Useage`, the model could be trained or saved using `torch. save`, but it …