Update modeling_aimv2.py
Browse files- modeling_aimv2.py +1 -0
modeling_aimv2.py
CHANGED
|
@@ -268,6 +268,7 @@ class AIMv2ForImageClassification(AIMv2PretrainedModel):
|
|
| 268 |
sequence_output = outputs[0]
|
| 269 |
|
| 270 |
logits = self.classifier(sequence_output[:, 0, :])
|
|
|
|
| 271 |
|
| 272 |
loss = None
|
| 273 |
if labels is not None:
|
|
|
|
| 268 |
sequence_output = outputs[0]
|
| 269 |
|
| 270 |
logits = self.classifier(sequence_output[:, 0, :])
|
| 271 |
+
print("LOGITS: ", logits)
|
| 272 |
|
| 273 |
loss = None
|
| 274 |
if labels is not None:
|