ccolas commited on
Commit
49cf77e
1 Parent(s): 4089f5b

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
@@ -63,6 +63,8 @@ def encoded2rep(encoded_path, rep_path=None, return_rep=False, verbose=False, le
63
  # rep = MODEL(chunk_tokenized)
64
  # reps.append(rep.detach().numpy())
65
  # representation = np.mean(reps, axis=0)
 
 
66
  representation = MODEL.encode(perf)
67
  error_msg += ' Nope. Saving performance?'
68
  np.savetxt(rep_path, representation)
 
63
  # rep = MODEL(chunk_tokenized)
64
  # reps.append(rep.detach().numpy())
65
  # representation = np.mean(reps, axis=0)
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)