Upload 106 files

fcclip/fcclip.py

@@ -181,9 +181,11 @@ class FCCLIP(nn.Module):
 
     def set_metadata(self, metadata):
         if set(self.test_metadata.stuff_classes) != set(metadata.stuff_classes):
+            print("setting test metadata:", metadata)
             self.test_metadata = metadata
             self.category_overlapping_mask, self.test_num_templates, self.test_class_names = self.prepare_class_names_from_metadata(metadata, self.train_metadata)
             self.test_text_classifier = None
+            print("text for classification:", self.test_class_names)
         return
 
     def get_text_classifier(self):