问题描述
使用YOLO或者其他框架训练的时候,出现了下面的问题
TypeError: Cannot handle this data type: (1, 1, 6), |u1
然后,你正在训练的数据就断了,这该怎么办? resume也不行了~~~
原因分析:
可以看下自己在训练的时候是不是使用了wandb这个功能。
什么?你不知道什么是wandb?
看下面的图,是不是很熟悉?如果你在训练的时候也出现了这个,那你就是用了wandb。
那么,就是wandb在做怪,具体是什么原因,有空更。把wandb解决了就可以正常运行了。
解决方案:
使用如下命令,卸载wandb
pip uninstall wandb
然后,使用resume功能继续训练就可以了。
注:等你什么时候又需要wandb了,就可以使用下面的安装命令,在安装上去就行了
pip install wandb
到这,就已经解决了问题。欢迎关注蓝胖胖,咱们一起解决BUG~~~