Guest User

Untitled

a guest
Oct 18th, 2017
67
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.75 KB | None | 0 0
  1. import tensorflow as tf
  2.  
  3. # define first network
  4. model_1 = tf.layers.dense(input1 , 100)
  5. ...
  6. loss_1 = ...
  7. summaries_1 = tf.summary.merge([tf.summary.scalar("loss_1", loss_1)])
  8. train_op_1 = ...
  9.  
  10.  
  11. # define second network
  12. model_2 = tf.layers.dense(input2 , 100)
  13. ...
  14. loss_2 = ...
  15. summaries_2 = tf.summary.merge([tf.summary.scalar("loss_2", loss_2)])
  16. train_op_2 = ...
  17.  
  18.  
  19. #define file writer
  20. fw = tf.summary.FileWriter(logdir='/tmp/my_logs')
  21.  
  22.  
  23. sess = tf.Session()
  24. # train your networks
  25.  
  26. for i in range(NUM_ITR):
  27.  
  28. # train first net
  29. _, summary_str = sess.run([train_op_1, summaries_1])
  30. fw.add_summary(summary_str, global_step=i)
  31.  
  32. # train second net
  33. _, summary_str = sess.run([train_op_2, summaries_2])
  34. fw.add_summary(summary_str, global_step=i)
Add Comment
Please, Sign In to add comment