Update encode_images_data.py

zjunlp · Feb 7, 2023 · 3e24c1a · 3e24c1a
1 parent c7802cc
commit 3e24c1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MarT/tools/encode_images_data.py b/MarT/tools/encode_images_data.py
@@ -25,7 +25,7 @@
         piexel_values = processor(images=images, return_tensors='pt')['pixel_values'].squeeze()
         entity2visual[entity] = piexel_values
 
-    torch.save(entity2visual, 'dataset/MARS/entity_image_features_total.pth')
+    torch.save(entity2visual, 'dataset/MARS/entity_image_features.CLIP-VIT-16-32.pth')
 
 elif processor_name == 'vilt':
     entity2visual = []