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

ihanwen99/Model-Pruning-in-Deep-Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Final Project of Algorithm

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages