ZJU-lishuang / yolov5_prune

yolov5 prune,Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
559 stars 137 forks source link

剪枝后怎么进行 finetune 训练呢? #60

Closed zimuyuan closed 3 years ago

zimuyuan commented 3 years ago

剪枝后网络的结构发生了变化,但是在 finetune 时要重新搭建网络结构需要对应网络的 .cfg 配置文件,例如 给出示例的命令中的 prune_0.5_keep_0.01_8x_yolov5s_v4_hand.cfg 但是这个文件怎么得到呢?自己剪枝的网络肯定和这个给出的不一样,是自动剪枝完成根据剪枝的结果自动生成的吗?没有看到剪枝后有对应额的文件生成。还是要自己手动修改的?