From 33b91086dc7b0748bedab5ba7e5b0d479adc1a32 Mon Sep 17 00:00:00 2001 From: Evening Date: Wed, 21 Feb 2024 13:13:50 +0800 Subject: [PATCH] Fix incorrect loss name index --- tests/integration_tests/test_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration_tests/test_pipeline.py b/tests/integration_tests/test_pipeline.py index a15fba6e..befe509c 100644 --- a/tests/integration_tests/test_pipeline.py +++ b/tests/integration_tests/test_pipeline.py @@ -42,5 +42,5 @@ def test_manual_segmentation_pipeline(ds): trainer = pl.Trainer(fast_dev_run=True) trainer.fit(m, datamodule=dm) - val_loss = trainer.validate(m, datamodule=dm)[0]["val_loss"] + val_loss = trainer.validate(m, datamodule=dm)[0]["val/ce_loss"] logging.debug(f"Validation score: {val_loss:.2%}")