maxmustermann/animegan2-pytorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

typo fixed

Browse source
This commit is contained in:
sr9 2021-02-17 14:03:56 +09:00
parent 00ee65231b
commit 87c26ac968
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
convert_weights.py
View file

@ -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")