def main(params): # load the checkpoint checkpoint_path = params[\'checkpoint_path\'] max_images = params[\'max_images\'] print (\'loading checkpoin