From 87c26ac968d666ff92d9da3086ea60e07c46e165 Mon Sep 17 00:00:00 2001 From: sr9 Date: Wed, 17 Feb 2021 14:03:56 +0900 Subject: [PATCH] typo fixed --- convert_weights.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert_weights.py b/convert_weights.py index e064709..19a784c 100644 --- a/convert_weights.py +++ b/convert_weights.py @@ -20,7 +20,7 @@ def load_tf_weights(tf_path): with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, device_count = {'GPU': 0})) as sess: ckpt = tf.train.get_checkpoint_state(tf_path) - assert ckpt is not None and ckpt.model_checkpoint_path is not None, f"Failed to load checkpoint {checkpoint_dir}" + assert ckpt is not None and ckpt.model_checkpoint_path is not None, f"Failed to load checkpoint {tf_path}" saver.restore(sess, ckpt.model_checkpoint_path) print(f"Tensorflow model checkpoint {ckpt.model_checkpoint_path} loaded")