Skip to content

jmarrietar/ocular

Repository files navigation

ocular

Description

Train

SimCLR - TPU

python utils/download.py -d train_voets
python train/simclr/pre_train_simclr.py --data_dir train_voets --num_epochs 500 --resume-epochs 400  --num_workers 8
python train/simclr/pre_train_simclr.py --data_dir train_voets --num_epochs 300

Notebooks

Notebooks Importantes

Fine_Tune_SimCLR_Pytorch_XLA.ipynb - Notebopok que hace el Finetune del SimCLR

KD_DR.ipynb Utiliza el modelo Fine tuneado para hacer un Teacher Student

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published