Skip to content
/ GIST Public

(ACM MM24) This is the offical repository of GIST: Improving Parameter Efficient Fine Tuning via Knowledge Interaction.

License

Notifications You must be signed in to change notification settings

JCruan519/GIST

Repository files navigation

GIST: Improving Parameter Efficient Fine Tuning via Knowledge Interaction

This repo is the official implementation of the Gist framework.

Usage

Install

  • Create a conda virtual environment and activate it:
conda create -n gist python=3.8 -y
conda activate gist
  • Install requirements:
pip install -r requirements.txt

Data preparation

  • VTAB-1K

You can follow ssf ("Scaling & Shifting Your Features: A New Baseline for Efficient Model Tuning") to download them.

Pre-trained model preparation

  • For pre-trained ViT-B/16 on ImageNet-21K, the model weights will be automatically downloaded. You can also manually download them from ViT.

Fine-tuning a pre-trained model via SSF

To fine-tune a pre-trained ViT model via Adapter within our GIST framework on VTAB-1K, run:

bash train_scripts/vit/train_vtab.sh

Acknowledgement

The code is built upon ssf.

About

(ACM MM24) This is the offical repository of GIST: Improving Parameter Efficient Fine Tuning via Knowledge Interaction.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published