NVlabs / NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
https://arxiv.org/abs/2007.03898
Other
999 stars 163 forks source link

Query: Dataset CelebA-HQ 256x256 issue #38

Open KomputerMaster64 opened 1 year ago

KomputerMaster64 commented 1 year ago

When i run the folllowing command to store the CelebA-HQ 256x256 dataset in an LMDB dataset, python convert_tfrecord_to_lmdb.py --dataset=celeba --tfr_path=$DATA_DIR/celeba/celeba-tfr --lmdb_path=$DATA_DIR/celeba/celeba-lmdb --split=train

I am getting the following error: Traceback (most recent call last): File "convert_tfrecord_to_lmdb.py", line 12, in <module> from tfrecord.torch.dataset import TFRecordDataset ModuleNotFoundError: No module named 'tfrecord'

I found only the following github issue relevant to the query: https://github.com/vahidk/tfrecord/issues/1

I request you to please guide me.