- 博客(2)
- 收藏
- 关注
原创 yolov5训练时出现_pickle.UnpicklingError: STACK_GLOBAL requires str的解决办法
在使用YOLOv5进行训练时,可能会出现_pickle.UnpicklingError: STACK_GLOBAL requires str 这样的错误,解决办法是,将数据集中的labels.cache文件删掉,如下图:在进行训练就好了,一般更换工程进行训练时,都需要将这个文件删掉才可以正常训练。...
2021-09-12 10:43:38
12854
7
原创 如何将GPU中的list转移到CPU中
在学习过程中遇到一个问题,就是当我想把一个GPU中的数据通过python中的serial模块传出来时,报错,并显示说我不能把GPU里面的数据拿出来,需要先将其转移到CPU中,当我按照他的提示使用 变量.CPU() 还是报错,提示我list没有 .cpu 属性。解决办法:使用torch.tensor()这个函数例如:L是一个在GPU中的list(列表)。torch.tensor(L, device=‘cpu’) 就可以将list转换为tensor,并将数据转移到cpu中。自己的是这么做的,肯定
2021-07-14 15:14:58
5899
8
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人