Commit cc7cad6e authored by kreuzer's avatar kreuzer
Browse files

Aktualisieren models.py

parent 2ad3bbeb

models.py

+0 −3
Original line number Diff line number Diff line
@@ -83,9 +83,6 @@ class SummarisationModel(nn.Module): 
                    print("Warning in Testing! This datapoint has an empty document or an empty summary")

                    continue



                if self.gpu: 

                    datapoint = datapoint.to(torch.device("cuda"))



                top_indices, probs = self.__call__(datapoint.document.to(self.device))

                r_1, r_2, r_l = utils.rouge(utils.select_elements(datapoint.raw_document, top_indices), datapoint.raw_summary, verbose=True)  

                running_rouge_1 += r_1