Added batch size feature for bAbI with RN

AndreaCossu / Relation-Network-PyTorch

Implementation of Relation Network and Recurrent Relational Network using PyTorch v1.3. Original papers: (RN) https://arxiv.org/abs/1706.01427 (RRN): https://arxiv.org/abs/1711.08028

MIT License

19 stars 7 forks source link

Added batch size feature for bAbI with RN #5

Closed AndreaCossu closed 5 years ago

AndreaCossu commented 5 years ago

It is now possible (and mandatory) to use batch_size > 1 with Relation Networks. This dramatically improves performances.