fix typos

IntelLabs · Nov 14, 2019 · 6b09bc4 · 6b09bc4
1 parent 8b94115
commit 6b09bc4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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

diff --git a/distiller/apputils/image_classifier.py b/distiller/apputils/image_classifier.py
@@ -471,7 +471,7 @@ def save_collectors_data(collectors, directory):
 
 def load_data(args, fixed_subset=False, sequential=False, load_train=True, load_val=True, load_test=True):
     # create flag to indicate if model is inception to use proper image size
-    if args.arch in ['inception_v3, inceptionv3, inceptionv4', 'inceptionresnetv2']:
+    if args.arch in ['inception_v3', 'inceptionv3', 'inceptionv4', 'inceptionresnetv2']:
         is_inception = True
     else:
         is_inception = False