trainer.py

[ghost]

In trainer.py, you define: self.n_train_batches = ds.n_train / self.batch_size
What happens if ds.n_train is not a multiple of self.batch_size? Eg n_train=166 and batch_size=20, then n_train_batches=8. So we are not using the last 6 elements in the training set.

[adbrebs]

Yes, at the moment these 6 elements would be discarded. So it is better to choose n_train % batch_size = 0, either modofying n_train or batch_size.

I am going to fix it by compiling two C functions, one for a batch size and the other one for (n_train % batch_size)