'eye_att' referenced before assignment
lcyuanjiang opened this issue · 2 comments
lcyuanjiang commented
'eye_att' referenced before assignment
G-force78 commented
ER-NeRF/nerf_triplane/network.py
Line 288 in c8ce2ec
Add eye_att = torch.tensor([])
cacard commented
ER-NeRF/nerf_triplane/network.py
Line 288 in c8ce2ec
Add eye_att = torch.tensor([])
will new error:
File "/root/_ai/ER-NeRF/nerf_triplane/utils.py", line 730, in train_step
eye_mask = data['eye_mask'] # [B, N]
KeyError: 'eye_mask'