Error when performing mesh extraction

[Shrief Gabr]

The following error is generated when running python ddp_mesh_nerf.py --config configs/lk2/final.txt. Any ideas how to resolve it?

2023-02-12 15:56:18,287 [INFO] root: Found ckpts: ['logs/lk2_final_masked_flipxzinitenv/model_420000.pth']
2023-02-12 15:56:18,287 [INFO] root: Reloading from: logs/lk2_final_masked_flipxzinitenv/model_420000.pth
Traceback (most recent call last):
  File "ddp_mesh_nerf.py", line 102, in <module>
    mesh()
  File "ddp_mesh_nerf.py", line 97, in mesh
    join=True)
  File "/projappl/project_2007011/miniconda3/envs/nerfosr/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 200, in spawn
    return start_processes(fn, args, nprocs, join, daemon, start_method='spawn')
  File "/projappl/project_2007011/miniconda3/envs/nerfosr/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 158, in start_processes
    while not context.join():
  File "/projappl/project_2007011/miniconda3/envs/nerfosr/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 119, in join
    raise Exception(msg)
Exception: 

-- Process 0 terminated with the following error:
Traceback (most recent call last):
  File "/projappl/project_2007011/miniconda3/envs/nerfosr/lib/python3.6/site-packages/torch/multiprocessing/spawn.py", line 20, in _wrap
    fn(i, *args)
  File "/scratch/project_2007011/NeRF-OSR/ddp_mesh_nerf.py", line 32, in ddp_mesh_nerf
    start, models = create_nerf(rank, args)
  File "/scratch/project_2007011/NeRF-OSR/ddp_train_nerf.py", line 420, in create_nerf
    models[name].load_state_dict(to_load[name])
  File "/projappl/project_2007011/miniconda3/envs/nerfosr/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1045, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DistributedDataParallel:
	Missing key(s) in state_dict: "module.env_params.01-08_07_30_IMG_6645-JPG", "module.env_params.01-08_07_30_IMG_6650-JPG",