Hi, thanks for releasing the code. However, how can I train "the best model" myself? Such a configuration is missing in hyperparams.py (simple_params does not correspond to it and makes many random choices anyway). Also, one cannot fully reconstruct it from best_vae_model.json, especially regarding optimization details.