Guest User

Untitled

a guest
Jun 22nd, 2018
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.04 KB | None | 0 0
  1. def draw_line(vis, step, lines, names):
  2. # draw multiple lines in one panel.
  3. """
  4.  
  5. :param vis: the object of the visdom.Visdom
  6. :param step: the step of the line
  7. :param lines: the lines tuple, (line1, line2, ...)
  8. :param names: the names tuple, (name1, name2, ...)
  9. :return: None
  10. """
  11. if not len(lines) == len(names):
  12. raise ValueError('The length of the input is not the same')
  13.  
  14. win_name = ''
  15. for i in range(len(names)):
  16. win_name += names[i]
  17.  
  18. if step == 0:
  19. for line, name in zip(lines, names):
  20. vis.line(X=torch.Tensor([step]),
  21. Y=torch.Tensor([line]),
  22. win=win_name,
  23. name='%s' % name,
  24. opts=dict(legend=[name])
  25. )
  26. else:
  27. for line, name in zip(lines, names):
  28. vis.updateTrace(X=torch.Tensor([step]),
  29. Y=torch.Tensor([line]),
  30. win=win_name,
  31. name='%s' % name
  32. )
Add Comment
Please, Sign In to add comment