shape (n_epochs,)

don't forget the , for a tuple with 1 element