yolov7遇到的bug

遇到的一些bug

使用的原有配置YOLOv5的pytorvh环境，版本如下：

ubuntu=20.04
NVIDIA-SMI 460.106.00 Driver Version: 460.106.00 CUDA Version: 11.2
pytorch=1.7.0
1.test==

_pickle.UnpicklingError: STACK_GLOBAL requires str

解决: 参考办法
我直接把yolo5里面的data.yaml复制到yolo7里面，但是路径地下有两个缓存文件–train.cache,val.cache ,把这俩文件删掉就可以了

2.train

RuntimeError: CUDA out of memory. Tried to allocate 2.96 GiB (GPU 0; 11.76 GiB total capacity; 1.31 GiB already allocated; 2.96 GiB free; 7.25 GiB reserved in total by PyTorch)

解决办法：参考方法
nvidia-smi
taskkill -PID 7392 -F(我电脑上没装taskkill)
kill 进程号
然后把训练命令的batch改为8，worker改为1