Biggest batch first during training as a sanity check

[vikigenius]

In older versions of allennlp. I distinctly remember an option "biggest_batch_first" that allowed a kind of sanity check where the biggest batch would be put first so that OOM errors are detected early.

Right now I could not find that anymore. Is there a way to replicate that behavior easily now? Can we bring back that feature? I found it very useful.