From c5e8630e500f4afe27d82385f58d9f6f08e62664 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 2 Aug 2021 08:12:07 +0100 Subject: [PATCH] lsun docs --- torchvision/datasets/lsun.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/torchvision/datasets/lsun.py b/torchvision/datasets/lsun.py index 75b284b597f..592806d30d0 100644 --- a/torchvision/datasets/lsun.py +++ b/torchvision/datasets/lsun.py @@ -55,8 +55,10 @@ def __len__(self) -> int: class LSUN(VisionDataset): - """ - `LSUN `_ dataset. + """`LSUN `_ dataset. + + You will need to install the ``lmdb`` package to use this dataset: run + ``pip install lmdb`` Args: root (string): Root directory for the database files.