人工智能
文章平均质量分 94
yushaochen
这个作者很懒,什么都没留下…
展开
-
NVIDIA TX2移植Caffe深度学习框架
文章目录1. 依赖顺序2. TX2 刷机2.1 NVIDIA-SDK-MANAGER 下载2.2 Nvidia TX2 SDK更新2.3 CUDNN测试3. caffe库移植3.1 修改Makefile.config3.2 修改caffe代码3.3 Crash修复3.3.1 cuda error code 7013.4 Caffe环境验证 1. 依赖顺序 Caffe <— CUDNN <— CUDA 2. TX2 刷机 2.1 NVIDIA-SDK-MANAGER 下载 官网下载,访问慢。需要注原创 2021-08-10 13:32:32 · 388 阅读 · 0 评论 -
通过子类来创建新的层和模型 making new layers and models via subclassing
文章目录通过子类来创建新的层和模型 making new layers and models via subclassing1. 准备2. 层类:将状态(权重参数)和一些计算进行组合,即为层类3. 层可具有不可训练的权重4. 最佳实践:延时创建权重,延时至知道输入的形状5. 层可递归组合6. add_loss()\fcolorbox{red}{lightgray}{add\_loss()}add_loss() 方法7. add_metric()\fcolorbox{red}{lightgray}{add\翻译 2021-08-07 19:40:43 · 159 阅读 · 0 评论