def save_model()

in tensorflow_managed_spot_training_checkpointing/source_dir/cifar10_keras_main.py [0:0]

• 15 lines of code
• 1 McCabe index (conditional complexity)

def save_model(model, output):
    signature = tf.saved_model.signature_def_utils.predict_signature_def(
        inputs={'image': model.input}, outputs={'scores': model.output}
    )

    builder = tf.saved_model.builder.SavedModelBuilder(output+'/1/')
    builder.add_meta_graph_and_variables(
        sess=K.get_session(),
        tags=[tf.saved_model.tag_constants.SERVING],
        signature_def_map={
            tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY:
                signature
        },
    )

    builder.save()
    logging.info("Model successfully saved at: {}".format(output))