task argument description

microsoft · Apr 5, 2022 · 1b040f9 · 1b040f9
1 parent c4393c1
commit 1b040f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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