winglian's picture
fix batch size calculation
5a631b3