diff --git a/torchgeo/datasets/millionaid.py b/torchgeo/datasets/millionaid.py index 043a6a7ab74..12e36115ca9 100644 --- a/torchgeo/datasets/millionaid.py +++ b/torchgeo/datasets/millionaid.py @@ -202,7 +202,7 @@ def __init__( Args: root: root directory where dataset can be found - task: whether to use multi-class or multi-label task + task: type of task, either "multi-class" or "multi-label" split: train or test split transforms: a function/transform that takes input sample and its target as entry and returns a transformed version