python run.py --model vae --config config.yaml python -m src.inference.inference --config imputation_config.yaml