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'
|
| 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),
|