Update cinic10.py

6 years ago · 7028443c88
1 changed files with 1 additions and 1 deletions
--- a/cinic10.py
+++ b/cinic10.py
@ -9,7 +9,7 @@ import tarfile

 def loadData(pathToDatasetFolder, oneHot=False):
    """
-    pathToDatasetFolder: Parent folder of CINIC-10 dataset folder of CINIC-10.tar.gz file
+    pathToDatasetFolder: Parent folder of CINIC-10 dataset folder or CINIC-10.tar.gz file
    oneHot: Label encoding (one hot encoding or not)

    Return: Train, validation and test sets and label numpy arrays