666DZY666 / micronet

micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference)、Low-Bit(≤2b)/Ternary and Binary(TWN/BNN/XNOR-Net); post-training-quantization(PTQ), 8-bit(tensorrt); 2、 pruning: normal、regular and group convolutional channel pruning; 3、 group convolution structure; 4、batch-normalization fuse for quantization. deploy: tensorrt, fp32/fp16/int8(ptq-calibration)、op-adapt(upsample)、dynamic_shape
MIT License
2.2k stars 478 forks source link

resume无法加载的问题 #55

Closed zhangfeixiang222 closed 3 years ago

zhangfeixiang222 commented 3 years ago

大神你好,关于IAO量化,我想问一下 self.register_buffer('scale', None) self.register_buffer('zero_point', None) 只有在forword之后才会分配相关的内存(我猜的,因为不训练没办法resume,提示这两个名称有误),可有解决的办法?多谢!