KeyError: 'network_fn_d_state_dict'

[AIBUWAN]

Fixing random seed 1
factor 2
(270, 480, 3, 12)
(270, 480, 12)
(270, 480, 12)
(270, 480, 2, 12)
(270, 480, 12)
Loaded ./data/Balloon2/ 45.061882503348485 71.07477444180932
Loaded llff (12, 270, 480, 3) (60, 3, 5) [270. 480. 418.96216] ./data/Balloon2/
DEFINING BOUNDS
NEAR FAR 0.0 1.0
Found ckpts ['./logs/Balloon2_H270_DyNeRF_pretrain/300000.tar', './logs/Balloon2_H270_DyNeRF_pretrain/Pretrained_S.tar']
Reloading from ./logs/Balloon2_H270_DyNeRF_pretrain/Pretrained_S.tar
Traceback (most recent call last):
File "run_nerf.py", line 768, in
train()
File "run_nerf.py", line 215, in train
render_kwargs_train, render_kwargs_test, start, grad_vars, optimizer = create_nerf(args)
File "/home/chenghuan/DynamicNeRF/folder/run_nerf_helpers.py", line 355, in create_nerf
model_d.load_state_dict(ckpt['network_fn_d_state_dict'])
KeyError: 'network_fn_d_state_dict'

Hi, thanks for your great work.

I wonder why there is no 'network_fn_d_state_dict' in ckpt.

Your model training time is slow . Can you take advantage of instant ngp-related work to increase speed?

我很疑惑为什么在我运行时ckpt中没有'network_fn_d_state_dict',您的模型训练时间很慢,我们能否利用 instant ngp 相关工作来提升您的模型的速度.期待您的回复.

[MasatoKawai]

If you just want to turn the code around, change ckpt_path in line 345 of run_nerf_helpers.py as follows

ckpt_path = ckpts[-2]

Before the change, "Pretrained_S" is loaded, but if you want to load other models, you can specify them yourself.

[AIBUWAN]

@MasatoKawai I just want to load "Pretrained_S", but it dosen't work.