Torch-Pruning
Torch-Pruning copied to clipboard
VainF
[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs
 Hello! Thank you very much for your excellent work! I wonder if I can ask you a question? It's that I got an error in the picture when I...
I'm not a yolo expert. But this line may be helpful for post-training: ```python pruned_macs, pruned_nparams = tp.utils.count_ops_and_params(pruner.model, example_inputs) print(model.model) print("Before Pruning: MACs=%f G, #Params=%f M" % (base_macs / 1e9,...
Description: Hi, I would like to prune a model developed using the MONAI library, but I am facing the following error: Code: # Define the model architecture network = UNet(...
Hello. I have made good use of the pruning scripts for YOLOv8, and was wondering if there will be a new examples script for YOLOv10? I am happy to collaborate...
Hello, recently I have been applying your code to yolov8. Based on previous problems encountered, and the help of the files in the examples/yolov8 folder. I managed to reduce both...
Not sure if this is intended behavior, but it looks like there might be an issue with concatenation based on the following test. Code: ``` class TestModule(nn.Module): def __init__(self, in_dim):...
