Why use this restrict? How to do when batch size is small on my machine?

[haowang1992]

https://github.com/ChenRocks/UNITER/blob/80d3602d71d65700eab373acb0507e31e251b7e7/data/sampler.py#L41

[badeaadi]

Hi @haowang1992 ,

This line tries to add more self._size_mul to the current batch it is creating. If your batch size is small and you have memory left, then you can increase it while keeping in mind that it represents the number of features from the examples, not the actual examples.

This issue might help you