Query: Dataset CelebA-HQ 256x256 issue

[Prabhav Kaula]

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: vahidk/tfrecord#1

I request you to please guide me.