Collect all batches, along with pred and loss, into self.data. Mainly for testing
CollectDataCallback(...) CollectDataCallback(...)
...
arguments to pass
None