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