How to use dataset using different keys

[errorfixrepeat]

I'm trying to use Flash for multilabel image data, but am having issues using ImageClassificationData.from_datasets(). The datasets are provided to me and I am unable to change their implementation.

Accessing an individual item (dataset[i]) returns a dictionary with the keys ('image','label') rather that ('input','target') used by Flash.

Is there somewhere I can override these defaults, or a hook where I can provide a function to change the dictionary keys?