Skip to content

How can i use torch pruning to ultralytics yolo model? #505

@hjy3451

Description

@hjy3451

Hi developer.

I want to use torch pruning to ultralytics’s yolov11n-segmentation model.

Previously, when I applied your torch pruning to other torch models, it performed well without any problems.

However, if the yolo model of ultralytics is imported into model = YOLO ('yolov11n.pt ') and torch pruning is applied, current_speed_up does not increase and does not increase from 1.

When I analyzed it, I think torch pruning can not fine conv2d in models of ultralytics.

So, how can i use torch pruning to yolov11?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions