HankYe / PAGCP

PAGCP for the compression of YOLOv5
GNU General Public License v3.0
111 stars 11 forks source link

opt.part? #21

Open xiaoshuomin opened 3 months ago

xiaoshuomin commented 3 months ago

有个部分代码没有看明白,想咨询一下
if name == "main": opt = parse_opt() if opt.compression.lower() == 'backbone': opt.part = [f'model.{i}.' for i in range(10)] else: opt.part = [f'model.{i}.' for i in range(24)] main(opt) 在compress.py文件中opt.part代表什么含义?如果修改这个值会有什么影响吗?

HankYe commented 2 months ago

Hi, thanks for ur attention to our work, and sorry for the late response. Opt.part refers to the list of layers that are expected to be compressed during pruning. So it should be adjusted to adapt to specific models.