Leacb4 commited on
Commit
8b62bf6
·
verified ·
1 Parent(s): fdab37b

Upload evaluation/basic_test_generalized.py with huggingface_hub

Browse files
evaluation/basic_test_generalized.py CHANGED
@@ -253,7 +253,7 @@ if __name__ == '__main__':
253
  hier_img_Bs.append(image_emb_hier.squeeze(0).detach().cpu())
254
 
255
  results.append({
256
- 'hierarchy' "hierarchy",
257
  'color': color,
258
  'row_index': int(row_idx),
259
  'sim_txt_color_part': float(sim_txt_color_part),
 
253
  hier_img_Bs.append(image_emb_hier.squeeze(0).detach().cpu())
254
 
255
  results.append({
256
+ 'hierarchy': hierarchy,
257
  'color': color,
258
  'row_index': int(row_idx),
259
  'sim_txt_color_part': float(sim_txt_color_part),