Showing
1 changed file
with
17 additions
and
0 deletions
config.py
0 → 100644
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 |
-
Please register or login to post a comment