Skip to content

Instantly share code, notes, and snippets.

@itsuncheng
Created June 12, 2020 10:38
Show Gist options
  • Save itsuncheng/816dcbb1ac868b7cd1a93aa269f267be to your computer and use it in GitHub Desktop.
Save itsuncheng/816dcbb1ac868b7cd1a93aa269f267be to your computer and use it in GitHub Desktop.
train_loss_list, valid_loss_list, global_steps_list = load_metrics(destination_folder + '/metrics.pt')
plt.plot(global_steps_list, train_loss_list, label='Train')
plt.plot(global_steps_list, valid_loss_list, label='Valid')
plt.xlabel('Global Steps')
plt.ylabel('Loss')
plt.legend()
plt.show()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment