From 76ed8967d7f4409da65cdbc0e54935ccb19cdace Mon Sep 17 00:00:00 2001 From: szmazurek Date: Fri, 10 Jan 2025 22:43:10 +0100 Subject: [PATCH] Histology classification tested --- testing/test_lightning_components.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/testing/test_lightning_components.py b/testing/test_lightning_components.py index d27a93a76..0be3f61fa 100644 --- a/testing/test_lightning_components.py +++ b/testing/test_lightning_components.py @@ -768,7 +768,6 @@ def test_port_model_inference_classification_histology_2d(device): parameters["model"]["dimension"] = 2 # read and parse csv training_data, parameters["headers"] = parseTrainingCSV(file_for_Training) - print(f"HEADERS: {parameters['headers']}") parameters["model"]["num_channels"] = 3 parameters["model"]["architecture"] = "densenet121" parameters["model"]["norm_type"] = "none" @@ -817,4 +816,6 @@ def test_port_model_inference_classification_histology_2d(device): num_sanity_val_steps=0, ) trainer.fit(module, train_dataloader, val_dataloader) - # trainer.test(module, test_dataloader) + trainer.test(module, test_dataloader) + + trainer.predict(module, test_dataloader)