Add model_names back. Fixes pytorch#195

Author: desimone

imagenet/main.py

@@ -15,6 +15,9 @@
 import torchvision.datasets as datasets
 import torchvision.models as models
 
+model_names = sorted(name for name in models.__dict__
+    if name.islower() and not name.startswith("__")
+    and callable(models.__dict__[name]))
 
 parser = argparse.ArgumentParser(description='PyTorch ImageNet Training')
 parser.add_argument('data', metavar='DIR',