From 10c4e3e0bf4572d5b1456ae180db006c343edc53 Mon Sep 17 00:00:00 2001
From: You Zhiyuan <42168004+zhiyuanyou@users.noreply.github.com>
Date: Sat, 5 Aug 2023 16:35:05 +0800
Subject: [PATCH] fix(cfg): fix dataset bug

fix dataset bug: bedroom dataset is mistaken as horse dataset
---
 config.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/config.py b/config.py
index 98068e8..c9bed7d 100644
--- a/config.py
+++ b/config.py
@@ -291,7 +291,7 @@ def make_dataset(self, path=None, **kwargs):
                               image_size=self.img_size,
                               **kwargs)
         elif self.data_name == 'bedroom256':
-            return Horse_lmdb(path=path or self.data_path,
+            return Bedroom_lmdb(path=path or self.data_path,
                               image_size=self.img_size,
                               **kwargs)
         elif self.data_name == 'celebalmdb':