regarding keyerror

[choycoy]

Hello, thank you for sharing your amazing work. I have got a Keyerror so could you give me any help?

<(nhp) C:\Users\razr\anaconda3\envs\nhp\Neural_Human_Performer-main>set CUDA_VISIBLE_DEVICES=0 & python run.py --type visualize --cfg_file configs/performance.yaml virt_data_root data/zju_mocap rasterize_root data/zju_rasterization ratio 0.5 H 1024 W 1024 test_input_view "0,7,15" test_mode model_x_motion_x exp_name demo resume True test.epoch -1 exp_folder_name debug gpus "0,"
Using torchvision resnet18 encoder
Pretrained: True
C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torchvision\models_utils.py:209: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and will be removed in 0.15, please use 'weights' instead.
f"The parameter '{pretrained_param}' is deprecated since 0.13 and will be removed in 0.15, "
C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torchvision\models_utils.py:223: UserWarning: Arguments other than a weight enum or None for 'weights' are deprecated since 0.13 and will be removed in 0.15. The current behavior is equivalent to passing weights=ResNet18_Weights.IMAGENET1K_V1. You can also use weights=ResNet18_Weights.DEFAULT to get the most up-to-date weights.
warnings.warn(msg)
load model: data/trained_model\if_nerf\demo\latest.pth
C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torch\utils\data\dataloader.py:560: UserWarning: This DataLoader will create 16 worker processes in total. Our suggested max number of worker in current system is 12 (cpuset is not taken into account), which is smaller than what this DataLoader is going to create. Please be aware that excessive worker creation might get DataLoader running slow or even freeze, lower the worker number to avoid potential slowness/freeze if necessary.
cpuset_checked))
the results are saved at data/perform/demo/epoch_-1/debug
0%| | 0/2171 [01:04<?, ?it/s]
Traceback (most recent call last):
File "run.py", line 127, in
globals() for batch in tqdm.tqdm(data_loader):
File "C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\tqdm\std.py", line 1195, in iter
for obj in iterable:
File "C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torch\utils\data\dataloader.py", line 652, in next
data = self._next_data()
File "C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torch\utils\data\dataloader.py", line 1347, in _next_data
return self._process_data(data)
File "C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torch\utils\data\dataloader.py", line 1373, in _process_data
data.reraise()
File "C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torch_utils.py", line 461, in reraise
raise exception
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torch\utils\data_utils\worker.py", line 302, in _worker_loop
data = fetcher.fetch(index)
File "C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users\razr\anaconda3\envs\nhp\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users\razr\anaconda3\envs\nhp\Neural_Human_Performer-main\lib\datasets\light_stage\can_smpl_perform.py", line 314, in getitem
msks = self.get_mask(index)
File "C:\Users\razr\anaconda3\envs\nhp\Neural_Human_Performer-main\lib\datasets\light_stage\can_smpl_perform.py", line 280, in get_mask
K = self.Ks[human][nv].copy()
KeyError: 'zju_mocap\CoreView_387'>