深度学习
cx媛
代码虐我千百遍,我待代码如初恋
展开
-
TensorRT 5/6/7 加载onnx模型
TensorRT5/6加载并转换onnx:def onnx_2_trt(onnx_model_name,trt_model_name): with trt.Builder(G_LOGGER) as builder, builder.create_network() as network, trt.OnnxParser(network, G_LOGGER) as parser: builder.max_batch_size = 1024 builder.原创 2020-07-27 18:11:49 · 5703 阅读 · 8 评论 -
pytorch 预训练model加载问题”Unexpected key(s) in state_dict:“
预训练模型使用 model = ResNet50() model = nn.DataParallel(model).cuda()加载预训练模型是使用model = ResNet50().cuda()checkpoint = torch.load(args.net_cache)model.load_state_dict(checkpoint['state_...原创 2020-03-02 14:45:02 · 10138 阅读 · 3 评论 -
版本坑
1.pytorch与tensorrt问题:pytorch -> onnx -> tensorrtpytorch1.1转Onnx正常,且onnx TVM中可用,但是onnx转到tensorrt(6.1.)时生成的.trt model只有及k:解决:版本问题,pytorch1.1转的onnx 与tensorrt6不兼容,pytorch升级到1.2解决,其它pytorch版本尚未...原创 2020-02-25 10:25:27 · 184 阅读 · 0 评论 -
Batch Normalization 学习笔记
Batch Normalization 学习笔记转至:http://blog.csdn.net/hjimce/article/details/50866313作者:hjimce一、背景意义本篇博文主要讲解2015年深度学习领域,非常值得学习的一篇文献:《Batch Normalization: Accelerating Deep Network Training by Reducing Inte...转载 2018-02-28 17:39:31 · 135 阅读 · 0 评论