Tensorflow训练模型

    import tensorflow as tf
    from pathlib import Path
    
    def save(path='./model/transfer_learn'):
    saver = tf.train.Saver()
    saver.save(sess, path, write_meta_graph=False)
    
    def restore(path='./model/transfer_learn'):
    saver = tf.train.Saver()
    saver.restore(sess, path)
    
    def train(input):
    a = tf.assign_add(input, tf.ones(1))
    return a
    
    if __name__ == "__main__":
    a = tf.Variable(tf.zeros(1),name='a')
    model_file = Path('./model')
    
    with tf.Session() as sess:
        if model_file.exists():
            restore()
        else: