Imported from https://github.com/ZhihongShao/Planning-based-Hierarchical-Variational-Model