
深度学习
文章平均质量分 95
嘤熊豪杰
重启解决一切问题
展开
-
PyTorch自定义C++拓展算子、转换ONNX模型、ORT部署实现
我们先来看一下建立的PyTorch模型,这是一个三层一维卷积模型,最后接了线性层完成分类任务输出。可以看到将应有的激活函数替换为了LIFSpike,并且传入了thresh和beta两个参数,模型中的其他参数省略。以下是LIFSpike的python部分实现,完整代码请见SNN-HAR,这里为了方便理解把源码中的ZIF替换为了FastSigmoid。原创 2024-03-01 11:25:55 · 1987 阅读 · 3 评论 -
Jetson TX2 使用指南
主要介绍如何配置Jetson并在其上进行图像推理原创 2023-03-10 10:39:34 · 1452 阅读 · 0 评论 -
基于WSL2+Docker的TensorFlow2配置流程
本流程在Windows 11下进行,也适用于最新版本的Windows 10。基于Windows操作系统,对于日常使用和同时使用多个其他的开发环境更加友好,上手难度低;而基于Linux的深度学习,生态成熟且案例也更多,因此WSL2是一个非常不错的选择。同时本文兼顾了基于Cuda的深度学习开发。原创 2023-02-02 22:39:13 · 966 阅读 · 0 评论