深度学习
文章平均质量分 56
小新GSUNG0222
这个作者很懒,什么都没留下…
展开
-
TypeError: can‘t convert cuda:0 device type tensor to numpy.
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.原创 2021-12-25 23:22:19 · 4287 阅读 · 2 评论 -
[CenterNet_Lite]使用自己的数据集训练CenterNet_Lite,从安装到使用
1、安装CenterNet_Lite1.1 环境配置先给出源码git地址:https://github.com/yjh0410/CenterNet-Lite我的服务器的环境:ubuntu16.04 、CUDA Version 9.0.176、CUDNN 7.5.0我使用了py3.6,pytorch==1.2.0 与之匹配的torchvision==0.4.0首先创建了一个新的虚拟环境:conda create -n CenterNetLite python=3.6source.原创 2020-11-30 20:21:40 · 2721 阅读 · 12 评论