config.json

{
  "checkpoint": null,
  "use_sam_checkpoint": false,
  "use_vit_sam_neck": false,
  "use_vit": true,
  "image_embed_dim": 768,
  "embed_dim": 512,
  "image_size": 1024,
  "vit_patch_size": 16,
  "class_attention": false,
  "example_attention": true,
  "example_class_attention": false,
  "class_embedding_dim": null,
  "spatial_convs": 3,
  "encoder_attention_downsample_rate": 2,
  "decoder_attention_downsample_rate": 2,
  "classification_layer_downsample_rate": 8,
  "use_support_features_in_prompt_encoder": true,
  "fusion_transformer": "TwoWayTransformer",
  "few_type": "Prototype",
  "class_fusion": "sum",
  "transformer_keys_are_images": true,
  "transformer_feature_size": null,
  "class_encoder": {
    "name": "RandomMatrixEncoder",
    "bank_size": 100,
    "embed_dim": 512
  },
  "segment_example_logits": false,
  "dropout": 0.0,
  "binary": false,
  "custom_preprocess": true,
  "encoder": "vit_b"
}