issues
search
MegEngine
/
Models
采用MegEngine实现的各种主流深度学习模型
Other
303
stars
99
forks
source link
fix(quantization): fix the order of quantize and load checkpoint
#93
Closed
dc3671
closed
3 years ago
dc3671
commented
3 years ago
目前不支持直接加载quantized model,因为quantized Module的output scale并没有作为state存下来
就算支持,mode=quantized也难以使用,因为目前开启后是load一个quantized模型,而不是生成一个quantized模型并保存,意味着没有从0到1的过程(除了直接读官方提供的模型)
其实更合适的做法是把mode=quantized作为另一个维度的选项,比如args.convert=True/False,即是否转换成quantized Module,这里为了和以前的教程视频兼容,暂不做修改
zhouyizhuang-megvii
commented
3 years ago
LGTM, FYI @wjfwzzc
wjfwzzc
commented
3 years ago
LGTM.