├── README.md ├── data └── data_links.txt └── src ├── args.py ├── datasets ├── cars.py ├── cifar10.py ├── cifar100.py ├── common.py ├── dtd.py ├── eurosat.py ├── flowers.py ├── gtsrb.py ├── imagenet.py ├── imagenet100.py ├── mnist.py ├── pets.py ├── registry.py ├── resisc45.py ├── stl10.py ├── sun397.py ├── svhn.py └── templates.py ├── eval.py ├── figures ├── comparison.png ├── exp.png ├── figures.txt ├── main_table.png ├── neulig_overview.png └── neulig_train_pip.png ├── finetune_clean.py ├── heads.py ├── modeling.py ├── neulig_main.py ├── pgbar.py ├── task_vectors.py ├── tm_utils.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/README.md -------------------------------------------------------------------------------- /data/data_links.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/data/data_links.txt -------------------------------------------------------------------------------- /src/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/args.py -------------------------------------------------------------------------------- /src/datasets/cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/cars.py -------------------------------------------------------------------------------- /src/datasets/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/cifar10.py -------------------------------------------------------------------------------- /src/datasets/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/cifar100.py -------------------------------------------------------------------------------- /src/datasets/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/common.py -------------------------------------------------------------------------------- /src/datasets/dtd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/dtd.py -------------------------------------------------------------------------------- /src/datasets/eurosat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/eurosat.py -------------------------------------------------------------------------------- /src/datasets/flowers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/flowers.py -------------------------------------------------------------------------------- /src/datasets/gtsrb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/gtsrb.py -------------------------------------------------------------------------------- /src/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/imagenet.py -------------------------------------------------------------------------------- /src/datasets/imagenet100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/imagenet100.py -------------------------------------------------------------------------------- /src/datasets/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/mnist.py -------------------------------------------------------------------------------- /src/datasets/pets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/pets.py -------------------------------------------------------------------------------- /src/datasets/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/registry.py -------------------------------------------------------------------------------- /src/datasets/resisc45.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/resisc45.py -------------------------------------------------------------------------------- /src/datasets/stl10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/stl10.py -------------------------------------------------------------------------------- /src/datasets/sun397.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/sun397.py -------------------------------------------------------------------------------- /src/datasets/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/svhn.py -------------------------------------------------------------------------------- /src/datasets/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/datasets/templates.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/figures/comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/figures/comparison.png -------------------------------------------------------------------------------- /src/figures/exp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/figures/exp.png -------------------------------------------------------------------------------- /src/figures/figures.txt: -------------------------------------------------------------------------------- 1 | Some figures in the paper are saved here. 2 | -------------------------------------------------------------------------------- /src/figures/main_table.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/figures/main_table.png -------------------------------------------------------------------------------- /src/figures/neulig_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/figures/neulig_overview.png -------------------------------------------------------------------------------- /src/figures/neulig_train_pip.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/figures/neulig_train_pip.png -------------------------------------------------------------------------------- /src/finetune_clean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/finetune_clean.py -------------------------------------------------------------------------------- /src/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/heads.py -------------------------------------------------------------------------------- /src/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/modeling.py -------------------------------------------------------------------------------- /src/neulig_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/neulig_main.py -------------------------------------------------------------------------------- /src/pgbar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/pgbar.py -------------------------------------------------------------------------------- /src/task_vectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/task_vectors.py -------------------------------------------------------------------------------- /src/tm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/tm_utils.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LiQiiiii/Neural-Ligand/HEAD/src/utils.py --------------------------------------------------------------------------------