Upload modeling_mic21.py with huggingface_hub

modeling_mic21.py

@@ -121,7 +121,7 @@ class MIC21SummarizerModel(PreTrainedModel):
             target_tok = self.components["tokenizer"](titles, add_special_tokens=False, max_length=max_len+1, padding='max_length')
             loss = torch.nn.CrossEntropyLoss()(out_logits.permute((0,2,1)), torch.LongTensor(target_tok["input_ids"]).cuda())
                 #.cuda(self.out_device))
-            return {"loss": loss, "logits": logits}
+            return {"loss": loss, "logits": logits, "eval_loss": loss}
             
         return {"logits":out_logits}