doc update
[jacinto-ai/caffe-jacinto.git] / plot_loss.py
1 #!/usr/bin/env python
3 import sys
4 import argparse
5 try:
6     import seaborn as sns
7     sns.set(style="dark")
8 except ImportError:
9     pass
10 import matplotlib.pyplot as plt
12 from common_plot import parse_files, plot_loss
14 parser = argparse.ArgumentParser(description="Plot loss from log files.")
15 parser.add_argument('-v', dest='value_at_hover', action='store_true',
16     help="Display plot values at cursor hover")
17 parser.add_argument('-s', dest='separate', action='store_true',
18     help="plot each log separately, don't concatenate them")
19 parser.add_argument('log', nargs = '*', help = "list of log files.")
20 args = parser.parse_args()
22 data = parse_files(files=args.log, separate=args.separate)
23 plt = plot_loss(data=data, value_at_hover=args.value_at_hover)
25 plt.show()