File size: 736 Bytes
2f13aa0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a4e201d
2f13aa0
a4e201d
2f13aa0
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
decap_weights: null
prefix_size: 768 
linear_talk2dino: False 
support_memory_size: 0
dino_model: 'dinov2_vitb14_reg'
normalize: False
kkv_attention: False
use_talk2dino_project: False
clip_model_name: "ViT-B/16"


# nested config
viecap:
  clip_hidden_size: 768
  suffix: ViT-B16_t2d_
  project_length: 10
  temperature: 0.01
  top_k: 3
  threshold: 0.4
  language_model: 'gpt2'
  name_of_entities_text: coco_entities #vinvl_vgoi_entities
  files_path: 'weights/viecap_files/'
  prompt_ensemble: True
  weight_path: 'weights/viecap-talk2dino-coco_karpathy-0014.pt'
  using_hard_prompt: True
  soft_prompt_first: True
  only_hard_prompt: False
  using_greedy_search: True #if false, use beam search
  beam_width: 5
  text_prompt: None