- 博客(3)
- 收藏
- 关注
RSS订阅原创 torch.cat
1.yolov4torch.cat 拼接问题 （已解决）cat常规操作x=torch.cat([x1,x0],dim=1)功能：将两个张量按相应维度拼接cat非常规操作上图class类为yolov4中的spp结构此处的第一个features为一个列表，内含三个张量，如下图第二个features将features列表与x列表相拼接返回一个维度为（2,2048,13,13）的张量2为bacth size大小2048为通道数13*13为shape...
2021-09-03 11:14:10
217
2
原创 tensorflow-gpu报错
刚换了台笔记本电脑 ，在用TensorFlow-gpu训练模型时，一直报错，踩了大坑，特在此写下我的解决方法。因我在解决gpu运行不了时尝试过更换cuda版本，更换cudnn版本，更换tensorflow-gpu与keras版本，故报的错误也是乱七八糟。报的错误一：2021-08-09 21:04:53.637764: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynam..
2021-08-12 21:50:23
3679
13
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人