Error during training Torso NeRF caused by deprecated param "lambda_ambient"
QuantumLiu opened this issue · 0 comments
QuantumLiu commented
In radnerf_sr.yaml
param lambda_ambient
is deprecated, but in Torso config it still exists, cause error when loading state_dict
Loaded Head Model from checkpoints/motion2video_nerf/zao_head
Loaded state_dict of Head Model to the RADNeRFTorso Model
Traceback (most recent call last):
File "/root/GeneFacePlusPlus/utils/commons/trainer.py", line 150, in fit
self.run_single_process(self.task)
File "/root/GeneFacePlusPlus/utils/commons/trainer.py", line 205, in run_single_process
self.restore_weights(checkpoint)
File "/root/GeneFacePlusPlus/utils/commons/trainer.py", line 467, in restore_weights
getattr(task_ref, k).load_state_dict(v, strict=True)
File "/root/miniconda3/envs/geneface/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2041, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for RADNeRFTorsowithSR:
Missing key(s) in state_dict: "torso_individual_codes", "density_grid_torso", "torso_embedder.embeddings", "torso_embedder.offsets", "head_color_weights_encoder.0.weight", "head_color_weights_encoder.0.bias", "head_color_weights_encoder.2.weight", "head_color_weights_encoder.2.bias", "head_color_weights_encoder.4.weight", "head_color_weights_encoder.4.bias", "torso_deform_net.net.0.weight", "torso_deform_net.net.1.weight", "torso_deform_net.net.2.weight", "torso_canonicial_net.net.0.weight", "torso_canonicial_net.net.1.weight", "torso_canonicial_net.net.2.weight".
Unexpected key(s) in state_dict: "lambda_ambient".
Terminated