"python test.py" 这个过程出现了问题

[concaki-sh]

我已经通过百度网盘下载到您给的权重，并将其严格按照文件目录的要求放到了指定文件夹下，在终端python test.py时出现以下问题：
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/data/miniforge/envs/urnn/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
data = fetcher.fetch(index)
File "/data/miniforge/envs/urnn/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/data/miniforge/envs/urnn/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/media/新加卷/jry/U-RNN/src/lib/dataset/Dynamic2DFlood.py", line 181, in getitem
input_vars = self._prepare_input(event_data, event_dir)
File "/media/新加卷/jry/U-RNN/src/lib/dataset/Dynamic2DFlood.py", line 119, in _prepare_input
event_data["absolute_DEM"]).float() * 1000
KeyError: 'absolute_DEM'

[concaki-sh]

问题在于：src/lib/dataset/Dynamic2DFlood.py 这一文件中本该是：
absolute_DEM = torch.from_numpy(event_data["absolute_DSM"]).float() * 1000
写成了：
absolute_DEM = torch.from_numpy(event_data["absolute_DEM"]).float() * 1000

[holmescao]

问题在于：src/lib/dataset/Dynamic2DFlood.py 这一文件中本该是： absolute_DEM = torch.from_numpy(event_data["absolute_DSM"]).float() * 1000 写成了： absolute_DEM = torch.from_numpy(event_data["absolute_DEM"]).float() * 1000

你的处理是对的。 应该是数据集在整理过程这个DSM和DEM的命名没对齐。

[yhxiong25]

我已经通过百度网盘下载到您给的权重，并将其严格按照文件目录的要求放到了指定文件夹下，在终端python test.py时出现以下问题： KeyError: Caught KeyError in DataLoader worker process 0. Original Traceback (most recent call last): File "/data/miniforge/envs/urnn/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/data/miniforge/envs/urnn/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/data/miniforge/envs/urnn/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/media/新加卷/jry/U-RNN/src/lib/dataset/Dynamic2DFlood.py", line 181, in getitem input_vars = self._prepare_input(event_data, event_dir) File "/media/新加卷/jry/U-RNN/src/lib/dataset/Dynamic2DFlood.py", line 119, in _prepare_input event_data["absolute_DEM"]).float() * 1000 KeyError: 'absolute_DEM'

这里应该是DSM， 我查看这个字典的所有KEY，只有一个是absolute_DSM，所以作者这里应该写错了