Add guards for batch size finder (#334)

Commit

4 years ago

Add guards for batch size finder (#334) * Fix zero reached Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>