Hello.
I am trying to edit cook_spinach scene of neural_3d dataset using pretrained checkpoints. But I am getting tensor shape mismatch error [snap attached].

For the purpose of context, I have used prepare_data tool that you have provided. Following is the run_edit_nerf.sh script and config file.

Thanks in advance.