Hi,
I found a bug in the definition of the on_epoch_end method of the VisCallback class in C1W5 Lab2:
this line:
display_digits(X_test, predictions, y_test, epoch, n=self.display_freq)
should be:
display_digits(X_test, predictions, y_test, epoch, n=self.n_samples)
The code works only because the default values of display_freq and n are both 10.
Cheers,
Istvan