embedding_models/vae/constants.py

# dimenstion of image embedding 
Z_DIM = 128
# hidden dimensions for encoder model
ENC_HIDDEN_DIM = 16
# hidden dimensions for decoder model
DEC_HIDDEN_DIM = 64