论文复现 视频分类中，想借用实践作业里的baseline来修改，跑通train.py后遇到的问题

Traceback (most recent call last):
File "work/train.py", line 128, in
train(args)
File "work/train.py", line 72, in train
train_config = merge_configs(config, 'train', vars(args))
File "/home/aistudio/work/config.py", line 54, in merge_configs
sec_dict = getattr(cfg, sec.upper())
File "/home/aistudio/work/utils.py", line 18, in __getattr__
return self[key]
KeyError: 'TRAIN'

相比于原版本只修改了"work/train.py" line76的自定义模型，（程序都没跑到），不知道为什么config会出问题，是因为版本升级到1.8.0后的问题么？