lsun test classes fix

Author: soumith

torchvision/datasets/lsun.py

@@ -70,7 +70,10 @@ def __init__(self, db_path, classes='train',
         dset_opts = ['train', 'val', 'test']
         self.db_path = db_path
         if type(classes) == str and classes in dset_opts:
-            classes = [c + '_' + classes for c in categories]
+            if classes == 'test':
+                classes = [classes]
+            else:
+                classes = [c + '_' + classes for c in categories]
         if type(classes) == list:
             for c in classes:
                 c_short = c.split('_')