pytorch量化感知训练(QAT)示例---ResNet
Posted 踟蹰横渡口,彳亍上滩舟。
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch量化感知训练(QAT)示例---ResNet相关的知识,希望对你有一定的参考价值。
pytorch量化感知训练(QAT)示例---ResNet
- 训练浮点模型,测试浮点模式在CPU和GPU上的时间;
- BN层融合,测试融合前后精度和结果比对;
- 加入torch的量化感知API,训练一个QAT模型;
- 保存定点INT8模型, 测试速度和精度;
- 完成一致性对其,并保存int8模型。
完整代码下载地址:下载地址
代码流程图下:
`
def main():
random_seed = 0
num_classes = 10
cuda_device = torch.device("cuda:0")
cpu_device = torch.device("cpu:0")
model_dir = "saved_models"
model_filename = "resnet18_cifar10.pt"
quantized_model_filename = "resnet18_quantized_cifar10.pt"
model_filepath = os.path.join(model_dir, model_filename)
quantized_model_filepath = os.path.join(
model_dir, quantized_model_filename)
set_random_seeds(random_seed=random_seed)
# Create an untrained model.
model = create_model(nu
以上是关于pytorch量化感知训练(QAT)示例---ResNet的主要内容,如果未能解决你的问题,请参考以下文章
基于pytorch量化感知训练(mnist分类)--浮点训练vs多bit后量化vs多bit量化感知训练效果对比
基于pytorch的模型剪枝+模型量化+BN合并+TRT部署(cifar数据)
基于pytorch的模型剪枝+模型量化+BN合并+TRT部署(cifar数据)
基于pytorch的模型剪枝+模型量化+BN合并+TRT部署(cifar数据)