├── .gitignore ├── LICENSE ├── README.md ├── configs ├── barlow_twins_grad_mm.py ├── byol_directpred_grad.py ├── simclr_grad_mm.py └── unigrad.py ├── figs └── comparison.png ├── lineval.py ├── pretrain.py ├── project ├── __init__.py ├── dataloader.py ├── model.py ├── pretrainer.py └── utils.py ├── run_linear.sh └── run_pretrain.sh /.gitignore: -------------------------------------------------------------------------------- 1 | build/ 2 | data/ 3 | exp/ 4 | 5 | __pycache__/ 6 | .ipynb_checkpoints/ -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/README.md -------------------------------------------------------------------------------- /configs/barlow_twins_grad_mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/configs/barlow_twins_grad_mm.py -------------------------------------------------------------------------------- /configs/byol_directpred_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/configs/byol_directpred_grad.py -------------------------------------------------------------------------------- /configs/simclr_grad_mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/configs/simclr_grad_mm.py -------------------------------------------------------------------------------- /configs/unigrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/configs/unigrad.py -------------------------------------------------------------------------------- /figs/comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/figs/comparison.png -------------------------------------------------------------------------------- /lineval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/lineval.py -------------------------------------------------------------------------------- /pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/pretrain.py -------------------------------------------------------------------------------- /project/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /project/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/project/dataloader.py -------------------------------------------------------------------------------- /project/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/project/model.py -------------------------------------------------------------------------------- /project/pretrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/project/pretrainer.py -------------------------------------------------------------------------------- /project/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/project/utils.py -------------------------------------------------------------------------------- /run_linear.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/run_linear.sh -------------------------------------------------------------------------------- /run_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fundamentalvision/UniGrad/HEAD/run_pretrain.sh --------------------------------------------------------------------------------