2024年12月9日
pytorch模型层的增、删、改
摘要： print打印下模型，可以看到各层的名字（小括号里的是层名） 如果层名是数字，写成 model.model[0].act 来定位到act层 import torch import ultralytics from ultralytics import YOLO yolo = YOLO("yolo11 阅读全文
posted @ 2024-12-09 15:44 夕西行 阅读(13) 评论(0) 推荐(0) 编辑