README.md

MedMNIST_selfsup

Dependency

install byol_pytorch from https://github.com/lucidrains/byol-pytorch

train_byol.py

Self-Supervised pretraining

train_imagenet.py

for non-self-supervised pretraining

train_supervised.py

for supervised training. Both fine-tuning and Linear evaluation are available by parsing --if-linear