Thank you for sharing this code. I want to train on a single modality dataset. Do you have any idea how should I change this code?