[Discussion] Exporting a checkpoint with static-sized tensors

deezer / spleeter

Deezer source separation library including pretrained models.

https://research.deezer.com/projects/spleeter.html

MIT License

25.74k stars 2.82k forks source link

[Discussion] Exporting a checkpoint with static-sized tensors #478

Open tinoucas opened 4 years ago

tinoucas commented 4 years ago

I'm trying to convert the 4stems checkpoint to TensorFlow-lite and I need the model to have static-sized tensors (so as to be able to use the Metal or CoreML delegates). I've tried specifying the input shape of 'waveform' but the outputs are still dynamic-sized.

Can this be done with the current checkpoints?

rodgomesc commented 1 year ago

@tinoucas have you made any progress on this? i'm also willing to dive in this challenge, i need to run spleeter models on android however it seems like none of the delegates support dynamic sized tensors, and the model have 174 dynamic tensors 😅