train.py

import os
import tensorflow as tf
import driving_data
import model

LOGDIR = './save'

sess = tf.InteractiveSession()

loss = tf.reduce_mean(tf.square(tf.sub(model.y_, model.y)))
train_step = tf.train.AdamOptimizer(1e-4).minimize(loss)
sess.run(tf.initialize_all_variables())

saver = tf.train.Saver()

#train over the dataset about 30 times
for i in range(int(driving_data.num_images * 0.3)):
  xs, ys = driving_data.LoadTrainBatch(100)
  train_step.run(feed_dict={model.x: xs, model.y_: ys, model.keep_prob: 0.8})
  if i % 10 == 0:
    xs, ys = driving_data.LoadValBatch(100)
    print("step %d, val loss %g"%(i, loss.eval(feed_dict={
         model.x:xs, model.y_: ys, model.keep_prob: 1.0})))
  if i % 100 == 0:
    if not os.path.exists(LOGDIR):
            os.makedirs(LOGDIR)
    checkpoint_path = os.path.join(LOGDIR, "model.ckpt")
    filename = saver.save(sess, checkpoint_path)
    print("Model saved in file: %s" % filename)