generation/util/util.py
separate_vae/util/util.py