Batch size - Githubissues

tristandeleu / ntm-one-shot

One-shot Learning with Memory-Augmented Neural Networks

MIT License

421 stars 94 forks source link

Batch size #2

Closed tristandeleu closed 8 years ago

tristandeleu commented 8 years ago

Add the ability to change the size of the batches
Changed the generator and metric to accept batch_size