Spaces:
Runtime error
Runtime error
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'])
|