Custom DataBunch

Hi,

I’m working on a project where every epoch I want to grab one item from each of my two classes, run them through two different models and then compare their output in a loss function. I am new to creating custom features in Fastai.

I am trying to start by creating a custom DataBunch that will output one image from each class every batch. I am a bit lost on where to start. Any help would be appreciated!

Thanks