diff --git a/README.md b/README.md index 52a3dc3..e891af1 100644 --- a/README.md +++ b/README.md @@ -9,6 +9,8 @@ XTrain, YTrain, XVal, YVal, XTest, YTest = cinic10.loadData("/path/to/dataset/fo `loadData()` method returns 6 numpy arrays respectively: Training set images, training set labels, validation set images, validation set labels, testing set images, testing set labels. +This script also downloads and extracts dataset files if dataset is not in the specified folder. + You can check `residual_network_cinic10.py` file for an example.