mini-batchの自由度を上げる

[imaharu]

現在の処理では、train / batch_size が割り切れない時にエラーを吐いてしまう