def test(self):
    test_loss = test(self.model, self.device, self.test_dataloader)
    return test_loss
def test(self):
    test_loss = test(self.model, self.device, self.test_dataloader)
    return test_loss