Denis Alekseenko
Denis Alekseenko
@luoshiyong Hello! Same error, please let us know if you find any solution
@chiraag-u Hello! Have you solved doublesized model after pruning?
@glenn-jocher > @tjasmin111 hey! 👋 Yes, selecting layers 7-15 for a 50% reduction sounds like a solid strategy to start with. It should give you a good balance between maintaining...
@glenn-jocher Here is how I do it ``` from ultralytics import YOLO import torch from torch.nn.utils import prune model = YOLO('yolov8m.pt') # check weights of original model print(model.model.model[0].conv.weight) print(model.model.model[0].conv.bias) ```...