def METHOD_NAME(self):
    model_for_classification = build_model(self.classification_config)
    model_for_classification.eval()
    model_for_classification = model_for_classification.to(get_current_device())
    with torch.no_grad():
        model_output = model_for_classification(self.sample_list)
    self.assertTrue("losses" in model_output)
    self.assertTrue("ce" in model_output["losses"])