line 179 i think losses should be inference_losses
line 179 i think losses should be inference_losses