- 博客(3)
- 资源 (2)
- 收藏
- 关注
原创 Pytorch、ONNX以及ONNX-TensorRT5.1 customOP注册
在自己注册TensorRT Op时发现出现错误如下:Unsupported ONNX data type: UINT8 (2)ERROR: /home/bokyliu/Work/TensorRT-5.1.5.0/onnx-tensorrt-5.1/ModelImporter.cpp:615 In function importModel:[8] Assertion failed: conve...
2020-01-19 17:44:46 2635 2
原创 caffe2与pytorch计算速度比较
测试结果模型:mobilenet-v2devicecaffe2pytorchcuda90ms8mscpu24ms10ms附import onnximport datetime# Load the ONNX modelmodel = onnx.load("model/mobilenet-v2_100.onnx")# Check that th...
2020-01-16 17:48:33 1076
原创 onnx-tensorrt5.1 ResizeNearest.cu代码解读
ResizeNearest.hpp:/* * Copyright (c) 2018, NVIDIA CORPORATION. All rights reserved. * * Permission is hereby granted, free of charge, to any person obtaining a * copy of this software and associa...
2020-01-02 09:43:03 846
k3s-airgap-images-arm64.tar
2020-05-14
k3s-arm64版本
2020-05-14
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人