ValueError: Invalid checkpoint state loaded from
JohnG0024 opened this issue · 0 comments
JohnG0024 commented
Hi, if I use new rendered 2d image dataset, I should just go through the last two bulletpoint commands in section Camera parameters estimation network
right?
But I'm getting
Traceback (most recent call last):
File "cam_est/train_sdf_cam.py", line 814, in <module>
train()
File "cam_est/train_sdf_cam.py", line 483, in train
ckptstate = tf.train.get_checkpoint_state(PRETRAINED_MODEL_PATH)
File "/home/johnG/anaconda3/envs/tf_1.13.0/lib/python3.6/site-packages/tensorflow/python/training/checkpoint_management.py", line 278, in get_checkpoint_state
+ checkpoint_dir)
ValueError: Invalid checkpoint state loaded from
when running this command
python -u cam_est/train_sdf_cam.py --log_dir cam_est/checkpoint/{your checkpoint dir} --gpu 0 --loss_mode 3D --learning_rate 1e-4 --src_h5_dir {your new rendered images' h5 directory} --img_h 224 --img_w 224
Any idea how to solve this?