Update src/music2cocktailrep/training/latent_translation/setup_trained_model.py

src/music2cocktailrep/training/latent_translation/setup_trained_model.py

@@ -32,6 +32,7 @@ def setup_trained_model(checkpoint_path=TRANSLATION_VAE_CHKP_PATH):
                                latent_dim=params['latent_dim'],
                                nb_classes=params['nb_classes'],
                                dropout=params['dropout'])
+    print('HERREE: ', filename, list(model.parameters())[0].sum())
     stats = params['stats']
     stats_music = np.array(stats['mean_std_music_rep'])
     stats_cocktail = np.array(stats['mean_std_cocktail_rep_norm11'])