Just a Typo (Lightning-Universe#413)

missing a ' at the end of dataset='stl10
chris-clem · Dec 9, 2020 · 969ceb3 · 969ceb3
1 parent 4d14bf4
commit 969ceb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/self_supervised_models.rst b/docs/source/self_supervised_models.rst
@@ -431,7 +431,7 @@ To Train::
     model = SwAV(
         gpus=1,
         num_samples=dm.num_unlabeled_samples,
-        dataset='stl10,
+        dataset='stl10',
         batch_size=batch_size
     )