├── .gitignore ├── LICENSE ├── README.md ├── README_zh.md ├── asset ├── feature.png ├── method.png └── prompts.png ├── attention.py ├── configs ├── cifar100_preprompt_5e.py ├── cub_preprompt_5e.py ├── five_datasets_preprompt_5e.py └── imr_preprompt_5e.py ├── continual_datasets ├── continual_datasets.py └── dataset_utils.py ├── datasets.py ├── engines ├── __init__.py └── prepromtp_engine.py ├── main.py ├── peft └── prompt │ └── pre_prompt.py ├── requirements.txt ├── trainers └── preprompt_trainer.py ├── training_scripts ├── train_5datsets_vit.sh ├── train_cifar100_vit.sh ├── train_cub_vit.sh └── train_imr_vit.sh ├── utils.py └── vits └── preprompt_vision_transformer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/README.md -------------------------------------------------------------------------------- /README_zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/README_zh.md -------------------------------------------------------------------------------- /asset/feature.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/asset/feature.png -------------------------------------------------------------------------------- /asset/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/asset/method.png -------------------------------------------------------------------------------- /asset/prompts.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/asset/prompts.png -------------------------------------------------------------------------------- /attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/attention.py -------------------------------------------------------------------------------- /configs/cifar100_preprompt_5e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/configs/cifar100_preprompt_5e.py -------------------------------------------------------------------------------- /configs/cub_preprompt_5e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/configs/cub_preprompt_5e.py -------------------------------------------------------------------------------- /configs/five_datasets_preprompt_5e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/configs/five_datasets_preprompt_5e.py -------------------------------------------------------------------------------- /configs/imr_preprompt_5e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/configs/imr_preprompt_5e.py -------------------------------------------------------------------------------- /continual_datasets/continual_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/continual_datasets/continual_datasets.py -------------------------------------------------------------------------------- /continual_datasets/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/continual_datasets/dataset_utils.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/datasets.py -------------------------------------------------------------------------------- /engines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /engines/prepromtp_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/engines/prepromtp_engine.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/main.py -------------------------------------------------------------------------------- /peft/prompt/pre_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/peft/prompt/pre_prompt.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/requirements.txt -------------------------------------------------------------------------------- /trainers/preprompt_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/trainers/preprompt_trainer.py -------------------------------------------------------------------------------- /training_scripts/train_5datsets_vit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/training_scripts/train_5datsets_vit.sh -------------------------------------------------------------------------------- /training_scripts/train_cifar100_vit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/training_scripts/train_cifar100_vit.sh -------------------------------------------------------------------------------- /training_scripts/train_cub_vit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/training_scripts/train_cub_vit.sh -------------------------------------------------------------------------------- /training_scripts/train_imr_vit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/training_scripts/train_imr_vit.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/utils.py -------------------------------------------------------------------------------- /vits/preprompt_vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/libo-huang/preprompt/HEAD/vits/preprompt_vision_transformer.py --------------------------------------------------------------------------------