yucornetto commited on
Commit
4b98e8d
1 Parent(s): 2df50c2

Upload 106 files

Browse files
Files changed (1) hide show
  1. fcclip/fcclip.py +2 -0
fcclip/fcclip.py CHANGED
@@ -181,9 +181,11 @@ class FCCLIP(nn.Module):
181
 
182
  def set_metadata(self, metadata):
183
  if set(self.test_metadata.stuff_classes) != set(metadata.stuff_classes):
 
184
  self.test_metadata = metadata
185
  self.category_overlapping_mask, self.test_num_templates, self.test_class_names = self.prepare_class_names_from_metadata(metadata, self.train_metadata)
186
  self.test_text_classifier = None
 
187
  return
188
 
189
  def get_text_classifier(self):
 
181
 
182
  def set_metadata(self, metadata):
183
  if set(self.test_metadata.stuff_classes) != set(metadata.stuff_classes):
184
+ print("setting test metadata:", metadata)
185
  self.test_metadata = metadata
186
  self.category_overlapping_mask, self.test_num_templates, self.test_class_names = self.prepare_class_names_from_metadata(metadata, self.train_metadata)
187
  self.test_text_classifier = None
188
+ print("text for classification:", self.test_class_names)
189
  return
190
 
191
  def get_text_classifier(self):