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

src/music2cocktailrep/training/latent_translation/setup_trained_model.py

@@ -33,8 +33,6 @@ def setup_trained_model(checkpoint_path=TRANSLATION_VAE_CHKP_PATH):
                                latent_dim=params['latent_dim'],
                                nb_classes=params['nb_classes'],
                                dropout=params['dropout'])
-    print('HEREEE: ', torch.sum(torch.Tensor([param.sum() for param in list(model.parameters())])))
-    print('model hash: ', hashlib.md5(open(model_path, 'rb').read()).hexdigest())
     stats = params['stats']
     stats_music = np.array(stats['mean_std_music_rep'])
     stats_cocktail = np.array(stats['mean_std_cocktail_rep_norm11'])
@@ -44,7 +42,8 @@ def setup_trained_model(checkpoint_path=TRANSLATION_VAE_CHKP_PATH):
 
     model.load_state_dict(torch.load(model_path))
     model.eval()
-
+    print('HEREEE: ', torch.sum(torch.Tensor([param.sum() for param in list(model.parameters())])))
+    print('model hash: ', hashlib.md5(open(model_path, 'rb').read()).hexdigest())
     def denormalize_cocktail_output(output):
         return output * stats_cocktail[1] + stats_cocktail[0]