dvae/model_config.json

{
    "enc_depth": 8,
    "dec_depth": 8,
    "dim": 256,
    "output_dim": 48,
    "latent_dim": 32,
    "T": 16,
    "num_traj": 512,
    "n_layers": 1
}