Spaces:
Runtime error
Runtime error
Update src/music/pipeline/encoded2rep.py
Browse files
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.'
|