ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Windows support

Browse Source
This commit is contained in:
allegroai 2019-06-17 00:58:46 +03:00
parent 9cd5f0fc56
commit c1d856d871
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/tensorflow_mnist_with_summaries.py
View File

@ -34,7 +34,7 @@ from tensorflow.examples.tutorials.mnist import input_data
from trains import Task from trains import Task
FLAGS = None FLAGS = None
task = Task.init(project_name='examples', task_name='Tensorflow mnist with summaries') task = Task.init(project_name='examples', task_name='Tensorflow mnist with summaries example')
def train(): def train():
@ -166,11 +166,9 @@ def train():
print('Accuracy at step %s: %s' % (i, acc)) print('Accuracy at step %s: %s' % (i, acc))
else: # Record train set summaries, and train else: # Record train set summaries, and train
if i % 100 == 99: # Record execution stats if i % 100 == 99: # Record execution stats
run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
run_metadata = tf.RunMetadata() run_metadata = tf.RunMetadata()
summary, _ = sess.run([merged, train_step], summary, _ = sess.run([merged, train_step],
feed_dict=feed_dict(True), feed_dict=feed_dict(True),
options=run_options,
run_metadata=run_metadata) run_metadata=run_metadata)
train_writer.add_run_metadata(run_metadata, 'step%04d' % i) train_writer.add_run_metadata(run_metadata, 'step%04d' % i)
train_writer.add_summary(summary, i) train_writer.add_summary(summary, i)