| # 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 | |