Changing batch size for behaviour cloning

Hi, I’m wondering if there is a way to use different batch sizes when running behaviour cloning?

It seems that currently BC relies on JsonReader.next(), which reads line by line, and doesn’t not support changing batch sizes. The default batch size is the rollout_fragment_length when data is collected.

I think I figured it out. By setting rollout_fragment_length in marwil one can control the BC batch size.