GitHub

Requirements

conda install pytorch==2.1.0 torchvision==0.16.0 cudatoolkit=12.1 -c pytorch

Fine-tune pretrained models with hyper-paramters sweep to obtain ground-truth fine-tuning results

bash finetune.sh

python forward_feature.py --dataset cifar10

python disco_metric.py --dataset cifar10

python tw.py --dataset cifar10

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
datasets		datasets
models/group1		models/group1
README.md		README.md
disco_metric.py		disco_metric.py
finetune.py		finetune.py
finetune.sh		finetune.sh
forward_feature.py		forward_feature.py
get_dataloader.py		get_dataloader.py
rank_correlation.py		rank_correlation.py
tw.py		tw.py
utils.py		utils.py