Further clarification on num_classes comment

official/resnet/cifar10_main.py

@@ -138,7 +138,7 @@ class Cifar10Model(resnet.Model):
       data_format: Either 'channels_first' or 'channels_last', specifying which
         data format to use when setting up the model.
       num_classes: The number of output classes needed from the model. This
-        enables users to extend the same model to different datasets.
+        enables users to extend the same model to their own datasets.
     """
     if resnet_size % 6 != 2:
       raise ValueError('resnet_size must be 6n + 2:', resnet_size)

official/resnet/imagenet_main.py

@@ -141,7 +141,7 @@ class ImagenetModel(resnet.Model):
       data_format: Either 'channels_first' or 'channels_last', specifying which
         data format to use when setting up the model.
       num_classes: The number of output classes needed from the model. This
-        enables users to extend the same model to different datasets.
+        enables users to extend the same model to their own datasets.
     """
 
     # For bigger models, we want to use "bottleneck" layers