ccolas commited on
Commit
004e4fc
1 Parent(s): 49cf77e

Update src/music/pipeline/encoded2rep.py

Browse files
Files changed (1) hide show
  1. src/music/pipeline/encoded2rep.py +2 -0
src/music/pipeline/encoded2rep.py CHANGED
@@ -66,6 +66,8 @@ def encoded2rep(encoded_path, rep_path=None, return_rep=False, verbose=False, le
66
  p = [int(p) for p in perf.split(' ')]
67
  print('PERF:', np.sum(p), perf)
68
  representation = MODEL.encode(perf)
 
 
69
  error_msg += ' Nope. Saving performance?'
70
  np.savetxt(rep_path, representation)
71
  error_msg += ' Nope.'
 
66
  p = [int(p) for p in perf.split(' ')]
67
  print('PERF:', np.sum(p), perf)
68
  representation = MODEL.encode(perf)
69
+ print('model weights sum: ', torch.sum(torch.Tensor([param.sum() for param in list(MODEL.parameters())])))
70
+ print('reprep', representation)
71
  error_msg += ' Nope. Saving performance?'
72
  np.savetxt(rep_path, representation)
73
  error_msg += ' Nope.'