# 1_feature_extractor
training information bottleneck

IF Model: https://huggingface.co/Qiyp/1_feature_extractor

Installation:
Clone the repository and then use the provided requirements.txt to install the dependencies:
pip install -r requirements.txt

put the proteus_vitb_backbone.pth into ./ckpt

load pretrained Feature Extractor for further training: --resume + Feature Extractor path

load pretrained Feature Extractor for feature extractraction: --finetune + Feature Extractor path

training scripts: 1_training_IB.sh
vis feature's semantic information: train_dec.sh