The [demo](https://github.com/pulkitag/ief/blob/master/src/test_demo.py#L74) only works for 1 example/batch at the moment. Upgrade it to work with batchsize > 1.
The demo only works for 1 example/batch at the moment. Upgrade it to work with batchsize > 1.