timeline test

eda/esmm/Model_pipline/train.py

@@ -362,12 +362,6 @@ def main(_):
     elif FLAGS.task_type == 'export':
         print("Not Implemented, Do It Yourself!")
 
-    tl = timeline.Timeline(run_metadata.step_stats)
-    ctf = tl.generate_chrome_trace_format()
-    with open(FLAGS.local_dir +'timeline.json', 'w') as wd:
-        wd.write(ctf)
-    print("耗时(分钟)：")
-    print((time.time()-b)/60)
 
 
 if __name__ == "__main__":
@@ -376,4 +370,12 @@ if __name__ == "__main__":
     ptions = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
     run_metadata = tf.RunMetadata()
     tf.logging.set_verbosity(tf.logging.INFO)
+    print("hello up")
     tf.app.run(options=ptions, run_metadata=run_metadata)
+    print("hello down")
+    tl = timeline.Timeline(run_metadata.step_stats)
+    ctf = tl.generate_chrome_trace_format()
+    with open(FLAGS.local_dir +'timeline.json', 'w') as wd:
+        wd.write(ctf)
+    print("耗时(分钟)：")
+    print((time.time()-b)/60)
\ No newline at end of file