- 博客(3)
- 收藏
- 关注
RSS订阅原创 CornerNet-Lite代码阅读
文章目录1. 数据准备2. 模型构建3. 损失函数与优化器4. 训练过程 之前在学习pytorch框架时候，有个老师说整个的使用过程，不外乎四个过程：1）数据准备（预处理、数据加载等）；2）模型定义；3）损失函数与优化器定义；4）进行训练 下面我将通过这样四个步骤对代码进行一个阅读。（划重点：自我理解！） 1. 数据准备 2. 模型构建 3. 损失函数与优化器 4. 训练过程 ...
2021-05-17 16:38:48
250
原创 多版本cuda运行与编译环境
多版本cuda运行与编译环境 基本环境： nvidi-smi后cuda版本为11.0 预装了两个版本的cuda分别为cuda-10.0和cuda-11.0 代码运行环境要求cuda-10.0 使用conda当前虚拟环境的cudatoolkit版本为10.0，代码运行一直报错，报错信息如下 cublas runtime error : the GPU program failed to execute at /opt/conda/condabld/pytorch_1544202130060/work/at
2020-12-29 22:19:44
1289
1
原创 CornerNet-Lite论文阅读
CornerNet-Lite论文阅读简介1. CornerNet-Saccdae2. CornerNet-Squeeze 论文地址：https://arxiv.org/abs/1904.08900 开源代码：https://github.com/princeton-vl/CornerNet-Lite 简介 CornerNet-Lite在CornerNet的基础之上，提出了轻量版，在尽量保证原检测精度的基础之上，通过两种方式提高检测的速度，以期达到实时监测的需求。CornerNet-Lite包含两个版本。 1
2020-09-20 15:48:08
555
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人