Trouble with loading the checkpoint

[ghost]

Hello ! @Joker316701882 I train the resface20 model with vggface2 dataset .And I want to restore the model checkpoint by the following code but it throw some error.

imagebatch = tf.placeholder(tf.float32,shape=(None,112,96,3),name='input') prelogits, = network.inference(image_batch, 1.0, phase_train=False, bottleneck_layer_size=512, weight_decay=0.0) embeddings = tf.nn.l2_normalize(prelogits, 1, 1e-10, name='embeddings') saver = tf.train.Saver() saver.restore(sess,tf.train.latest_checkpoint(save_path))

The error is:

NotFoundError: Key Resface/Bottleneck/BatchNorm/moving_mean not found in checkpoint [[Node: save/RestoreV2 = RestoreV2[dtypes=[DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, ..., DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT, DT_FLOAT], _device="/job:localhost/replica:0/task:0/device:CPU:0"](_arg_save/Const_0_0, save/RestoreV2/tensor_names, save/RestoreV2/shape_and_slices)]] [[Node: save/RestoreV2/_133 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device_incarnation=1, tensor_name="edge_138_save/RestoreV2", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"]()]]

I didn't change any code, and I train the model just following your readme.md instruction.

Please tell me how to deal with it , Thank you very much .

[Joker316701882]

@jijianyi Try to train again. I think the problem has been fixed.

[ghost]

Thank you ! it worked !