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

this works with minimal change

Browse Source
This commit is contained in:
Brandon Thomas
2020-06-02 19:45:36 -04:00
parent e5864ac8b3
commit 08e5d29e3d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
glow-tts-jaywalnut/wrap_glow_model.py
+1 -1
View File
@@ -172,7 +172,7 @@ if trace_model:
# NB: Getting `Tracing failed sanity checks! Graphs differed across invocations!` # NB: Getting `Tracing failed sanity checks! Graphs differed across invocations!`
inputs = (x_tst, x_tst_lengths) inputs = (x_tst, x_tst_lengths)
#traced_script_module = torch.jit.trace(glow_model, inputs, check_trace=True) #traced_script_module = torch.jit.trace(glow_model, inputs, check_trace=True)
traced_script_module = torch.jit.trace(glow_model, inputs, check_trace=False) traced_script_module = torch.jit.trace(glow_model, inputs, check_trace=True)
traced_script_module.save(output_filename) traced_script_module.save(output_filename)
else: else:
print('Saving script model as: {}'.format(output_filename)) print('Saving script model as: {}'.format(output_filename))