deezer / deezer/spleeter

[Discussion] Exporting a checkpoint with static-sized tensors

Open
#478 1 comment 2 reactions 0 assignees View on GitHub
question
Dominant language
Python
Stars
28.4k
Forks
3.1k
PR merge metrics
No merged PRs in 30d

Description

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?

Contributor guide

Open the contributing guide

Research direction

Start with the 4stems checkpoint conversion and the waveform input-shape setting described in the report. Check whether TensorFlow Lite export preserves static output shapes for the current checkpoints and record a reproducible result or limitation.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, tensorflow
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.