在使用mmdetection训练的时候出现了如下错误:
2022-05-07 21:35:13,488 - mmdet - INFO - Epoch [1][50/5111] lr: 1.978e-04, eta: 18:35:19, time: 0.546, data_time: 0.217, memory: 1781, loss_cls: 17.7358, loss_bbox: 4.1217, loss: 21.8
575
2022-05-07 21:35:24,322 - mmdet - INFO - Epoch [1][100/5111] lr: 3.976e-04, eta: 12:58:42, time: 0.217, data_time: 0.003, memory: 1781, loss_cls: 13.2629, loss_bbox: 4.4149, loss: 17.6
778
2022-05-07 21:35:35,155 - mmdet - INFO - Epoch [1][150/5111] lr: 5.974e-04, eta: 11:06:25, time: 0.217, data_time: 0.003, memory: 1781, loss_cls: 9.6519, loss_bbox: 4.5696, loss: 14.22
16
2022-05-07 21:35:45,901 - mmdet - INFO - Epoch [1][200/5111] lr: 7.972e-04, eta: 10:09:16, time: 0.215, data_time: 0.003, memory: 1781, loss_cls: 13.5727, loss_bbox: 5.6242, loss: 19.1
969
runner.run(data_loaders, cfg.workflow)
File "D:ProgramDataAnaconda3envsmytorchlibsite-packagesmmcv
unnerepoch_based_runner.py", line 127, in run
epoch_runner(data_loaders[i], **kwargs)
File "D:ProgramDataAnaconda3envsmytorchlibsite-packagesmmcv
unnerepoch_based_runner.py", line 51, in train
self.call_hook('after_train_iter')
File "D:ProgramDataAnaconda3envsmytorchlibsite-packagesmmcv
unnerase_runner.py", line 309, in call_hook
getattr(hook, fn_name)(self)
File "D:ProgramDataAnaconda3envsmytorchlibsite-packagesmmdet-2.24.1-py3.7.eggmmdetcorehookcheckloss_hook.py", line 24, in after_train_iter
runner.logger.info('loss become infinite or NaN!')
AssertionError: None
出现这个问题是因为学习率太大了,需要降低学习率;
把学习率改为2e-4就可以解决问题了。