先占个坑。
-
train时遇到错误:UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead
解决方法: https://blog.csdn.net/BBZZ2/article/details/101022935 -
用自己的数据集训练,改变了类别数。训练时会出现错误如下:
RuntimeError: CUDA error: device-side assert triggered.
目前找到的有效的解决方法有两个:-
在utils/utils.py中,在相应位置添加下面代码:
https://github.com/eriklindernoren/PyTorch-YOLOv3/issues/157
b, target_labels =
-