diff --git a/distiller/apputils/data_loaders.py b/distiller/apputils/data_loaders.py index ff4eeeaea23e0b98ad5fa7dad09af4bd34852f4b..2394aa9038a2b5d07c16695ac6374a12739ce328 100755 --- a/distiller/apputils/data_loaders.py +++ b/distiller/apputils/data_loaders.py @@ -71,7 +71,7 @@ def load_data(dataset, data_dir, batch_size, workers, validation_split=0.1, dete Args: dataset: a string with the name of the dataset to load (cifar10/imagenet) - data_dir: the directory where the datset resides + data_dir: the directory where the dataset resides batch_size: the batch size workers: the number of worker threads to use for loading the data validation_split: portion of training dataset to set aside for validation