训练自己数据集loss为nan

[hxy1051653358]

你好，我使用自己数据集进行训练，大概1000多张图片，五个类别，改了config.py中类别及data path 使用voc预训练权重，出现梯度爆炸是什么原因？ Traceback (most recent call last): File "train.py", line 197, in YoloTrain().train() File "train.py", line 143, in train raise ArithmeticError('The gradient is exploded') ArithmeticError: The gradient is exploded

[Stinky-Tofu]

你好，我使用自己数据集进行训练，大概1000多张图片，五个类别，改了config.py中类别及data path 使用voc预训练权重，出现梯度爆炸是什么原因？ Traceback (most recent call last): File "train.py", line 197, in YoloTrain().train() File "train.py", line 143, in train raise ArithmeticError('The gradient is exploded') ArithmeticError: The gradient is exploded

目前估计不支持训练自己的数据集，因为我没在其他数据集上跑过，所以可能会存在bug

[hxy1051653358]

@Stinky-Tofu 那你是在所有bn位置都换成了group normalization吗？还是如何使用的group normalization?

[Stinky-Tofu]

@Stinky-Tofu 那你是在所有bn位置都换成了group normalization吗？还是如何使用的group normalization?

用过，但是没效果，所以就没用加进去了，只是提供了GN的开源代码

[wizholy]

我在自己数据集上跑过，可以用

[HangyuQ]

wizholy，您好，方便加下微信交流下吗？hy1786591 ，现在在用自己数据集训练，谢谢！