yunjey

config for compatibility with tensorflow 0.11 and 0.12

Showing 1 changed file with 17 additions and 0 deletions
1 +import tensorflow as tf
2 +
3 +# configuration for tensorflow 0.11 and 0.12 version
4 +try:
5 + # tensorflow 0.12 version
6 + image_summary = tf.summary.image
7 + scalar_summary = tf.summary.scalar
8 + histogram_summary = tf.summary.histogram
9 + merge_summary = tf.summary.merge_all
10 + SummaryWriter = tf.summary.FileWriter
11 +except:
12 + # tensorflow <= 0.11 version
13 + image_summary = tf.image_summary
14 + scalar_summary = tf.scalar_summary
15 + histogram_summary = tf.histogram_summary
16 + merge_summary = tf.merge_all_summaries
17 + SummaryWriter = tf.train.SummaryWriter
...\ No newline at end of file ...\ No newline at end of file