├── LICENSE ├── README.md ├── SimDis ├── data │ ├── __init__.py │ ├── dataset.py │ └── transforms.py ├── exps │ └── arxiv │ │ ├── __init__.py │ │ ├── base_exp.py │ │ ├── base_exp_simDis.py │ │ ├── linear_eval_exp.py │ │ ├── linear_eval_exp_simDis.py │ │ └── tools.py ├── layers │ └── optimizer.py ├── models │ ├── __init__.py │ ├── resnet_mbn.py │ ├── sim_dis_eval_model.py │ └── sim_dis_train_model.py ├── tools │ ├── __init__.py │ └── run_SimDis.py └── utils │ ├── __init__.py │ ├── log.py │ └── torch_dist.py └── requirements.txt /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/README.md -------------------------------------------------------------------------------- /SimDis/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /SimDis/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/data/dataset.py -------------------------------------------------------------------------------- /SimDis/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/data/transforms.py -------------------------------------------------------------------------------- /SimDis/exps/arxiv/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /SimDis/exps/arxiv/base_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/exps/arxiv/base_exp.py -------------------------------------------------------------------------------- /SimDis/exps/arxiv/base_exp_simDis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/exps/arxiv/base_exp_simDis.py -------------------------------------------------------------------------------- /SimDis/exps/arxiv/linear_eval_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/exps/arxiv/linear_eval_exp.py -------------------------------------------------------------------------------- /SimDis/exps/arxiv/linear_eval_exp_simDis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/exps/arxiv/linear_eval_exp_simDis.py -------------------------------------------------------------------------------- /SimDis/exps/arxiv/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/exps/arxiv/tools.py -------------------------------------------------------------------------------- /SimDis/layers/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/layers/optimizer.py -------------------------------------------------------------------------------- /SimDis/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /SimDis/models/resnet_mbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/models/resnet_mbn.py -------------------------------------------------------------------------------- /SimDis/models/sim_dis_eval_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/models/sim_dis_eval_model.py -------------------------------------------------------------------------------- /SimDis/models/sim_dis_train_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/models/sim_dis_train_model.py -------------------------------------------------------------------------------- /SimDis/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /SimDis/tools/run_SimDis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/tools/run_SimDis.py -------------------------------------------------------------------------------- /SimDis/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/utils/__init__.py -------------------------------------------------------------------------------- /SimDis/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/utils/log.py -------------------------------------------------------------------------------- /SimDis/utils/torch_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/SimDis/utils/torch_dist.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JindongGu/SimDis/HEAD/requirements.txt --------------------------------------------------------------------------------