recognize-anything

Sleeping

xinyu1205 commited on Mar 10, 2023

Commit

217926f

•

1 Parent(s): ba7689a

Update models/tag2text.py

Files changed (1) hide show

models/tag2text.py CHANGED Viewed

@@ -92,7 +92,7 @@ class Tag2Text_Caption(nn.Module):
         self.threshold = threshold
         num_features = 768
-        self.num_class = config['class_num']
         q2l_config = BertConfig.from_json_file('configs/q2l_config.json')
         q2l_config.encoder_width = vision_width

         self.threshold = threshold
         num_features = 768
+        self.num_class = 3429
         q2l_config = BertConfig.from_json_file('configs/q2l_config.json')
         q2l_config.encoder_width = vision_width