CoolFace
Apppublic

Minho511/Compositional-Zero-Shot-Learning-Demo

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
0likes
pdt.yml36 linesDownload Raw Back to root
1---2dataset:3  dataset: cgqa4  splitname: compositional-split-natural5  train_only: true6  test_set: val7 8model_params:9  image_extractor: vit-dino10  emb_type: word2vec11  nlayer: 212  mlp_layers: [768]13  delta_encoder: [1024]14  delta_encoder_layers: 215  delta_decoder: [768]16  delta_decoder_layers: 217  18training:19  max_epochs: 5020  batch_size: 12821  update_image_features: false # Fine tune the image backbone?22  update_word_features: true23  extract_feature_vectors: true24  lr: 5.0e-0525  lr_image_extractor: 5.0e-0526  lr_word_embedder: 5.0e-0527  cosine_scale: 0.0528  lr_delta_encoder: 5.0e-0529  rank: 030 31eval:32  test_batch_size: 51233  topk: 134  alpha: 0.035exp:36  seed: 214