【施工中】torch2trt 学习笔记Sep 18, 2020·2 minsDeep-Learning Jetson Nano 模型转换前言 # 偶然发现了 torch2trt 的模型转换方案,思路是直接将pytorch op映射到TensorRT的python api. 在pytorch进行每个op forward的时候,tensorrt也相应往network上添加op. 这里会先涉及torch2trt的使用,后面会补充这个转换工具的代码学习
Kubernetes(k8s)在深度学习模型转换方面的探索Nov 22, 2019·4 mins工程 工程实践 模型转换 Docker K8s年中的时候接了离职的同事模型转换的锅,在不断地更新迭代的过程中,发现了一些痛点。 发现k8s能够解决一部分痛点,因此来分享一下在这方面的探索。