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

src/music2cocktailrep/training/latent_translation/setup_trained_model.py

@@ -32,7 +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())
+    print('HEREEE: ', torch.sum(torch.Tensor([param.sum() for param in list(model.parameters())])))
     stats = params['stats']
     stats_music = np.array(stats['mean_std_music_rep'])
     stats_cocktail = np.array(stats['mean_std_cocktail_rep_norm11'])