Just a Typo (Lightning-Universe#413)

missing a ' at the end of dataset='stl10
chris-clem · Dec 16, 2020 · 931dad8 · 931dad8
1 parent 07315ae
commit 931dad8
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
     )