如何训练tensorflow模型

class Model(object):

    def __init__(self,img_size ,num_layers,.feature_size,channels):

        self.input = tf.palceholder(tf.float32,[None,img_size,img_size,channels])

        self.output = tf.palceholder(tf.float32,[None,img_size,img_size,channels])

        self.loss =         

        tf.summary.scalar("loss",self.loss)

        tf.summary.image("input_image",tf.cast(self.input,tf.uint8))

        self.sess = tf.Session()

        self.saver = tf.train.Saver

    def train(self,iterations=1000，save_dir=“saved_models”)：

        if not os.path.exists(save_dir):

            os.mkdir(save_dir)

        merged = tf.summary.merge_all()

        optimizer = tf.train.AdamOptimizer(learning_rate= 0.001)

        train_op = optimizer.minimize(self.loss)

        init = tf.global_variables_initializer()

        with self.sess as sess:

            sess.run(init)

            self.saver.restore(self.sess,tf.train.latest_checkpoint(save_dir="saved_models"))

            self.saver.save(self.sess,save_dir="saved_models/model")