Size difference between umxhq and umxl

[sevagh]

Hello,

I'm curious how the size of the pth file for umxl and umxhq are different. Despite the larger data set, isn't it the same bilstm model, therefore the same total number of weights?

sevagh@pop-os:~/repos/umx.cpp$ du -h  ~/.cache/torch/hub/checkpoints/* | sort -h
# hq
34M     /home/sevagh/.cache/torch/hub/checkpoints/bass-8d85a5bd.pth
34M     /home/sevagh/.cache/torch/hub/checkpoints/drums-9619578f.pth
34M     /home/sevagh/.cache/torch/hub/checkpoints/other-b52fbbf7.pth
34M     /home/sevagh/.cache/torch/hub/checkpoints/vocals-b62c91ce.pth

# xl
108M    /home/sevagh/.cache/torch/hub/checkpoints/bass-2ca1ce51.pth
108M    /home/sevagh/.cache/torch/hub/checkpoints/drums-69e0ebd4.pth
108M    /home/sevagh/.cache/torch/hub/checkpoints/other-c8c5b3e6.pth
108M    /home/sevagh/.cache/torch/hub/checkpoints/vocals-bccbd9aa.pth

Actually I forgot to check the target .json files, which confirm the hidden size is bigger in UMXL.

[sevagh]

I'm closing this issue now, but I just wanted to show a few projects I worked on related to this question I asked:

• umx.cpp, load umxhq or umxl weights in a custom binary format, perform inference with custom C++ code that only relies on the standard library
• free-music-demixer, UMX + quantized UMX-L running in the browser with WebAssembly

[faroit]

@sevagh umx.cpp is super cool!