From 7fc74e099d26a8f6cb104464b7be436a3bf7b4fa Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sun, 25 Dec 2022 15:07:33 -0600 Subject: [PATCH] Add comment explaining mismatch --- torchgeo/datamodules/loveda.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchgeo/datamodules/loveda.py b/torchgeo/datamodules/loveda.py index b1fd8d78c6c..a82233f9844 100644 --- a/torchgeo/datamodules/loveda.py +++ b/torchgeo/datamodules/loveda.py @@ -77,6 +77,7 @@ def setup(self, stage: Optional[str] = None) -> None: split="val", transforms=val_predict_transforms, **self.kwargs ) + # Test set masks are not public, use for prediction instead self.predict_dataset = LoveDA( split="test", transforms=val_predict_transforms, **self.kwargs )