storytold/torchscript-models
2
0
Fork 0
mirror of https://github.com/storytold/torchscript-models.git synced 2026-10-18 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Melgan requires a sample mel for tracing

Browse Source
This commit is contained in:
Brandon Thomas
2020-06-02 02:25:23 -04:00
parent 72dd12b460
commit ce2be638f4
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
melgan-swpark/sample/LJ002-0320.mel
BIN
View File
Binary file not shown.
melgan-swpark/wrap_melgan_model.py
+1 -1
View File
@@ -30,7 +30,7 @@ output_filename = 'melgan_trump_txlearn_2020-05-30_steno_chkpt9800.torchjit'
trace_model = True
if trace_model:
print('Tracing model and saving as: {}'.format(output_filename))
mel_file = '/home/bt/dev/voder/data/mels/LJ002-0320.mel'
mel_file = './sample/LJ002-0320.mel'
mel = torch.load(mel_file, map_location=torch.device('cpu'))
# NB: Getting `Tracing failed sanity checks! Graphs differed across invocations!`
traced_script_module = torch.jit.trace(melgan_model, mel, check_trace=True)