Skip to content
This repository has been archived by the owner on Jul 18, 2021. It is now read-only.

Latest commit

 

History

History
24 lines (16 loc) · 556 Bytes

README.md

File metadata and controls

24 lines (16 loc) · 556 Bytes

Model-Pruning-in-Deep-Learning

Final Project of Algorithm


代码逻辑

本目录下:

  • train_vgg19.py 从预训练模型继续训练
  • prune_demo.py 直接使用 torch 库进行裁剪

codes 文件夹是自己剪枝方法的实现

本文使用的模型下载后放置到根目录下。


PDF格式报告链接: Report.pdf 1 2 3 4 5