Open hxy1051653358 opened 5 years ago
你好,我使用自己数据集进行训练,大概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
@Stinky-Tofu 那你是在所有bn位置都换成了group normalization吗?还是如何使用的group normalization?
@Stinky-Tofu 那你是在所有bn位置都换成了group normalization吗?还是如何使用的group normalization?
用过,但是没效果,所以就没用加进去了,只是提供了GN的开源代码
我在自己数据集上跑过,可以用
wizholy,您好,方便加下微信交流下吗?hy1786591 ,现在在用自己数据集训练,谢谢!
你好,我使用自己数据集进行训练,大概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