The following error occurs when training a model with custom data:
Traceback (most recent call last):
File "D:\CG\humannerf-main\train.py", line 32, in
main()
File "D:\CG\humannerf-main\train.py", line 25, in main
trainer.train(epoch=epoch,
File "core/train/trainers/human_nerf/trainer.py", line 146, in train
train_loss, loss_dict = self.get_loss(
File "core/train/trainers/human_nerf/trainer.py", line 104, in get_loss
rgb = net_output['rgb']
KeyError: 'rgb'"
The following error occurs when training a model with custom data:
Traceback (most recent call last): File "D:\CG\humannerf-main\train.py", line 32, in
main()
File "D:\CG\humannerf-main\train.py", line 25, in main
trainer.train(epoch=epoch,
File "core/train/trainers/human_nerf/trainer.py", line 146, in train
train_loss, loss_dict = self.get_loss(
File "core/train/trainers/human_nerf/trainer.py", line 104, in get_loss
rgb = net_output['rgb']
KeyError: 'rgb'"
May I ask where the problem is