Merge branch 'master' of https://github.com/YannDubs/disentangling-vae …

…into master
YannDubs · Feb 10, 2021 · 7b8285b · 7b8285b
2 parents f8f64d8 + c578558
commit 7b8285b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -370,6 +370,7 @@ def __init__(self, root=os.path.join(DIR, '../data/mnist'), **kwargs):
 class FashionMNIST(datasets.FashionMNIST):
     """Fashion Mnist wrapper. Docs: `datasets.FashionMNIST.`"""
     img_size = (1, 32, 32)
+    background_color = COLOUR_BLACK
 
     def __init__(self, root=os.path.join(DIR, '../data/fashionMnist'), **kwargs):
         super().__init__(root,