×
分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

Test zhangchenchen2021-09-16 10:55:26 回复 1 查看 技术答疑
Test
分享到:

if opt.mlu:
    model = torchvision.models.inception_v3().eval()
    # net = net.fuse()
    
    from torchvision.models.quantization.utils import quantize_model
    model = quantize_model(model)

    state_dict = torch.load("../../model/online/inceptionv3_int8.pth")
    model = torch_mlu.core.mlu_quantize.quantize_dynamic_mlu(model)
    model.load_state_dict(state_dict, strict=False)
    net = model
    import pdb;pdb.set_trace()
    net.to(ct.mlu_device())
    print('quantized_model load succeful ... ')

版权所有 © 2021 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭