- Maybe you can push [VAE-Pytorch](https://github.com/explainingai-code/VAE-Pytorch/tree/main)/[utils](https://github.com/explainingai-code/VAE-Pytorch/tree/main/utils/extract_mnist_images.py) to this repo too. - ModuleNotFoundError: No module named 'dataset' pops up. It happened in VAE code too. So to include `import sys sys.path.append("./") `