def draw_train_loss(loss_train, loss_valid, save_path):
    plt.plot(loss_train)
    plt.plot(loss_valid)
    plt.title('model loss')
    plt.ylabel('loss')
    plt.xlabel('epoch')
    plt.legend(['train', 'validation'], loc='upper left')
    plt.savefig(save_path)
    plt.close()
def draw_train_loss(loss_train, loss_valid, save_path):
    plt.plot(loss_train)
    plt.plot(loss_valid)
    plt.title('model loss')
    plt.ylabel('loss')
    plt.xlabel('epoch')
    plt.legend(['train', 'validation'], loc='upper left')
    plt.savefig(save_path)
    plt.close()