Expectation-Maximization via Pretext-Invariant Representations

IEEE Access 2023

Overview of our framework

---

Requirements

• torch
• torchvision
• tqdm
• einops
• wandb
• pytorch-lightning
• lightning-bolts
• torchmetrics
• scipy
• timm

---

Installation

Please refer to solo-learn repository.

---

Training

Please refer to bash_files folders. We provide bash files for both CIFAR and ImageNet datasets.

Star if you find it useful. ⭐