[Discussion] Exporting a checkpoint with static-sized tensors

[tinoucas]

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]

@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 😅