def plot_loss(loss_arr):
    l = plt.figure()
    plt.plot(loss_arr)
    plt.title('Loss')
    plt.ylabel('loss')
    plt.xlabel('epoch')
    plt.legend(plot_legend, loc='upper left')
    l.show()
def plot_loss(loss_arr):
    l = plt.figure()
    plt.plot(loss_arr)
    plt.title('Loss')
    plt.ylabel('loss')
    plt.xlabel('epoch')
    plt.legend(plot_legend, loc='upper left')
    l.show()