[scripts] fixed stuff related to model selection (and add the latent dim param...

[scripts] fixed stuff related to model selection (and add the latent dim param needed for the original architecture)