average loss = 0.319224