Final Project of Algorithm
本目录下:
train_vgg19.py
从预训练模型继续训练prune_demo.py
直接使用 torch 库进行裁剪
codes
文件夹是自己剪枝方法的实现
本文使用的模型下载后放置到根目录下。
PDF格式报告链接: Report.pdf
Final Project of Algorithm
本目录下:
train_vgg19.py
从预训练模型继续训练prune_demo.py
直接使用 torch 库进行裁剪codes
文件夹是自己剪枝方法的实现
本文使用的模型下载后放置到根目录下。
PDF格式报告链接: Report.pdf