| #!/usr/bin/env python3 | |
| import sys | |
| import matplotlib.pyplot as plt | |
| file_path = sys.argv[1] | |
| with open(file_path, "r") as f: | |
| lines = f.readlines() | |
| loss = [] | |
| key_word = "constrast_loss: " | |
| for line in lines: | |
| if key_word in line: | |
| loss.append(line.split(key_word)[-1].split("|")[0]) | |
| X = range(len(loss)) | |
| plt.plot(X, loss) | |
| plt.savefig(file_path + "_plot.png") | |