'eye_att' referenced before assignment

Fictionarry / ER-NeRF

[ICCV'23] Efficient Region-Aware Neural Radiance Fields for High-Fidelity Talking Portrait Synthesis

https://fictionarry.github.io/ER-NeRF/

MIT License

1.08k stars 138 forks source link

Open lcyuanjiang opened 6 months ago

G-force78 commented 6 months ago

Add eye_att = torch.tensor([])

cacard commented 5 months ago

https://github.com/Fictionarry/ER-NeRF/blob/c8ce2ec1689bb6050f5567c5dd66320d43cbe2c1/nerf_triplane/network.py#L288

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'