Documentation `n_batches_in_epoch` incorrect argument name

[liamcripwell]

Just letting you know in the README.md you reference an argument with the name n_batches_in_epoch but within the code this argument is actually called n_batches_per_epoch.

[IssamLaradji]

Thanks, it is now fixed :)