1、RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB
1、num_workers 调整为0
2、batchsize调整为1 ???
2、报错解决:ValueError: Media Bounding Box Debugger/Images is invalid. Please remove invalid filename characters
解决方法:
1、关闭wandb
在 yolov5\utils\loggers\wandb\wandb_utils.py
try:
import wandb
assert hasattr(wandb, '__version__') # verify package import not local dir
except (ImportError, AssertionError):
wandb = None
在此程序后面加一行:wandb = None
try:
import wandb
assert hasattr(wandb, '__version__') # verify package import not local dir
except (ImportError, AssertionError):
wandb = None
wandb = None
如果继续出现错误报告:AttributeError: 'WandbLogger' object has no attribute 'bbox_interval'
解决方法:在 \yolov5\utils\loggers 文件夹下的 _init__.py文件中里面的
try:
import wandb
assert hasattr(wandb, '__version__') # verify package import not local dir
except (ImportError, AssertionError):
wandb = None
下面再次添加一行wandb = None 变为如下所示,即可成功
try:
import wandb
assert hasattr(wandb, '__version__') # verify package import not local dir
except (ImportError, AssertionError):
wandb = None
wandb = None
√