update dataset mode

1 files changed, 4 insertions, 3 deletions

diff --git a/data/image_folder.py b/data/image_folder.py
index 44e15cb..898200b 100644
--- a/data/image_folder.py
+++ b/data/image_folder.py
@@ -1,9 +1,9 @@
-################################################################################
+###############################################################################
 # Code from
 # https://github.com/pytorch/vision/blob/master/torchvision/datasets/folder.py
 # Modified the original code so that it also loads images from the current
 # directory as well as the subdirectories
-################################################################################
+###############################################################################
 
 import torch.utils.data as data
 
@@ -45,7 +45,8 @@ class ImageFolder(data.Dataset):
         imgs = make_dataset(root)
         if len(imgs) == 0:
             raise(RuntimeError("Found 0 images in: " + root + "\n"
-                               "Supported image extensions are: " + ",".join(IMG_EXTENSIONS)))
+                               "Supported image extensions are: " +
+                               ",".join(IMG_EXTENSIONS)))
 
         self.root = root
         self.imgs = imgs