for epoch in range(epochs):
    epoch_losses = []