ViT-Dino for Medical images

HI!

I would like to thank you first for such a good and updated repo regarding Vision Transformers.

I want to know if I can use 3d medical images to pretrain the ViT using 3D medical images?. Do I need to make some changes to the sample code you shared.

Thanks