Hello,
I don't know if it's an error, or if i missunderstand, but in training_step method
loss = mean_squared_error(predictions, targets)
And loss method take params in wrong (i supposed) order
def mean_squared_error(targets, predictions):
In any case thanks a lot for the free book !