py3 compat

Author: soumith

requirements.txt

@@ -1 +1,2 @@
 pillow
+six

torchvision/datasets/lsun.py

@@ -2,7 +2,7 @@
 from PIL import Image
 import os
 import os.path
-import StringIO
+import six
 import string
 import sys
 if sys.version_info[0] == 2:
@@ -34,7 +34,7 @@ def __getitem__(self, index):
         with env.begin(write=False) as txn:
             imgbuf = txn.get(self.keys[index])
 
-        buf = StringIO.StringIO()
+        buf = six.BytesIO()
         buf.write(imgbuf)
         buf.seek(0)
         img = Image.open(buf).convert('RGB')