2ca285b61fb8280243a55b37d76297cd109328c8,chiron/chiron_eval_test.py,,,#,152

Before Change

    fig, ax1 = plt.subplots()
    ax2=ax1.twinx()
    x1=range(2000)
    x2=range(400)
    ax1.plot(x1,batch_x[INDEX,:],"black")
    for i in range(5):
        ax2.plot(x2,prob_val[INDEX,:,i],COLOR[i])        

After Change

    config.gpu_options.allow_growth = True
    saver = tf.train.Saver(var_list=tf.trainable_variables()+tf.moving_average_variables())
    sess = tf.train.MonitoredSession(session_creator=tf.train.ChiefSessionCreator(config=config))
    saver.restore(sess, tf.train.latest_checkpoint(FLAGS.model))
    if os.path.isdir(FLAGS.input):
        file_list = os.listdir(FLAGS.input)
        file_dir = FLAGS.input
    else:

Instances