ccolas commited on
Commit
da8c12a
1 Parent(s): 01538c1

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

Browse files
src/music2cocktailrep/training/latent_translation/setup_trained_model.py CHANGED
@@ -32,6 +32,7 @@ def setup_trained_model(checkpoint_path=TRANSLATION_VAE_CHKP_PATH):
32
  latent_dim=params['latent_dim'],
33
  nb_classes=params['nb_classes'],
34
  dropout=params['dropout'])
 
35
  stats = params['stats']
36
  stats_music = np.array(stats['mean_std_music_rep'])
37
  stats_cocktail = np.array(stats['mean_std_cocktail_rep_norm11'])
 
32
  latent_dim=params['latent_dim'],
33
  nb_classes=params['nb_classes'],
34
  dropout=params['dropout'])
35
+ print('HERREE: ', filename, list(model.parameters())[0].sum())
36
  stats = params['stats']
37
  stats_music = np.array(stats['mean_std_music_rep'])
38
  stats_cocktail = np.array(stats['mean_std_cocktail_rep_norm11'])