File size: 575 Bytes
1633fcc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# 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
|