Minibatching with FFJORD

The following should work?

train_loader = Flux.Data.DataLoader((train_data,), batchsize=10)