在训练时 显示没有定义这个字典里的key

[Liurongfen]

(DiffPhys) PS E:\LRF\UW-DiffPhys-main\UW-DiffPhys-main> python train_UW-DDIM.py --config underwater_lsui_uieb_128.yml --resume ckpts/LSUI_UIEB_ddpm.pth.tar
Using device: cuda
=> using dataset 'LSUI_UIEB'
Traceback (most recent call last):
File "E:\LRF\UW-DiffPhys-main\UW-DiffPhys-main\train_UW-DDIM.py", line 75, in
main()
File "E:\LRF\UW-DiffPhys-main\UW-DiffPhys-main\train_UW-DDIM.py", line 66, in main
DATASET = datasets.dict"Underwater"
KeyError: 'Underwater'

在训练时 显示没有定义这个字典里的key

[bachzz]

Could you try again with the updated dataset loader file