Add comment explaining mismatch

microsoft · Dec 25, 2022 · 05269be · 05269be
1 parent 940f4da
commit 05269be
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchgeo/datamodules/loveda.py b/torchgeo/datamodules/loveda.py
@@ -81,6 +81,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
         )