import tensorflow as tf

ckpt_name = 'save/model.ckpt'
fname = 'model.tf'
dst_nodes = ['output/predictions']
saver = tf.train.Saver()

# Create a session and init
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    
    print("training started!!")
    print("******************")

    # Now iterate over our dataset n_epoch times
    for epoch_i in range(epoch):
