上传者: 42121272
|
上传时间: 2022-01-17 17:17:57
|
文件大小: 847KB
|
文件类型: -
Dorefa网
的pytorch实施 .The代码的灵感来自和 。
要求
python> 3.5
火炬> = 1.1.0
火炬视觉> = 0.4.0
每晚,未来(用于张量板)
nvidia-dali> = 0.12(更快的 )
Cifar-10精度
从头开始训练量化模型
模型
W_位
一点点
加速器
网路18
32
32
94.71%
网路18
4
4
94.36%
网路18
1个
4
93.87%
ImageNet精度
从头开始训练量化模型
模型
W_位
一点点
前1名
前5
网路18
32
32
69.80%
89.32%
网路18
4
4
66.60%
87.15%
用法
下载ImageNet数据集并将验证图像移动到带标签的子文件夹中。为此,您可以使用以下
训练模型
python3 cifar_train_eval.py
pyt