Save a HUB model to a checkpoint

save_hub_checkpoint.py

import tensorflow as tf
import tensorflow_hub as hub

model_size = 'xlarge'.upper()
version = 2

model = hub.Module("https://tfhub.dev/google/albert_{}/{}".format(model_size.lower(), version), trainable=False)
init = tf.global_variables_initializer()
saver = tf.train.Saver()

with tf.Session() as sess:
    sess.run(init)
    path = "path_to_albert/albert"
    save_path = saver.save(sess, path)