深度学习框架
深度学习框架
HollrayChan
这个作者很懒,什么都没留下…
展开
-
【PyTorch】学习之路—简简单单的入门,一文看懂
Pytorch原创 2019-09-15 00:50:27 · 6479 阅读 · 3 评论 -
【TensorRT】自定义层插件plugin的使用—结合fc_plugin_caffe_mnist
【TensorRT】自定义层插件plugin的使用 官网中给了几个关于插件plugin的使用,只有现成的教程,比如: TensorRT&Sample&Python[fc_plugin_caffe_mnist] TensorRT学习笔记5 - 运行fc_plugin_caffe_mnist 但是,总有你预想不到的情况发生。 接下来将结合官网例子:samples/tensorRT_sa...原创 2019-12-30 18:47:35 · 2127 阅读 · 12 评论 -
【TensorRT】使用记录 —— TensorRTv6
TensorRT使用记录 TensorRT Developer Guide Python IConstantLayer method 框架介绍 TensorRT的流程:输入是一个预先训练好的FP32的模型和网络,将模型通过parser等方式输入到TensorRT中,TensorRT可以生成一个Serialization,也就是说将输入串流到内存或文件中,形成一个优化好的engine,执行的时候可...原创 2019-12-17 20:50:00 · 561 阅读 · 0 评论