update dataset mode

1 files changed, 12 insertions, 0 deletions

diff --git a/data/base_dataset.py b/data/base_dataset.py
new file mode 100644
index 0000000..49b9d98
--- /dev/null
+++ b/data/base_dataset.py
@@ -0,0 +1,12 @@
+import torch.utils.data as data
+
+class BaseDataset(data.Dataset):
+    def __init__(self):
+        super(BaseDataset, self).__init__()
+        
+    def name(self):
+        return 'BaseDataset'
+    
+    def initialize(self, opt):
+        pass
+