From 6fdbaa2ad18a39186d44ac4455314a0e4e6f50ec Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Sun, 5 Jan 2020 03:29:17 -0500 Subject: [PATCH] this works --- tacotron_melgan/src/server.rs | 2 +- tacotron_melgan/tacotron_model.py | 2 +- tacotron_melgan/wrap_tacotron2_model.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tacotron_melgan/src/server.rs b/tacotron_melgan/src/server.rs index ef7620e239..7824f2eda6 100644 --- a/tacotron_melgan/src/server.rs +++ b/tacotron_melgan/src/server.rs @@ -97,7 +97,7 @@ pub fn run_server() { .mount("/pub", StaticFiles::new("/www/public", options).rank(-1))*/ //let tacotron = "/home/bt/dev/voder/tacotron_melgan/tacotron_jit_model_voder_c0cac635.pt"; - let tacotron = "/home/bt/dev/voder/tacotron_melgan/tacotron_container2.pt"; + let tacotron = "/home/bt/dev/voder/tacotron_melgan/tacotron_container3.pt"; //let melgan = "/home/bt/dev/voder/tacotron_melgan/melgan_jit_model_voder_c0cac635.pt"; let melgan = "/home/bt/dev/voder/tacotron_melgan/melgan_container2.pt"; let melgan = "/home/bt/dev/tacotron-melgan/shared_melgan_container.pt"; diff --git a/tacotron_melgan/tacotron_model.py b/tacotron_melgan/tacotron_model.py index 566caef178..2f470da0a2 100644 --- a/tacotron_melgan/tacotron_model.py +++ b/tacotron_melgan/tacotron_model.py @@ -601,4 +601,4 @@ class Tacotron2(nn.Module): ]) #return outputs - return outputs[0] + return outputs[0].float() diff --git a/tacotron_melgan/wrap_tacotron2_model.py b/tacotron_melgan/wrap_tacotron2_model.py index 7c02f611a7..c65070e76b 100755 --- a/tacotron_melgan/wrap_tacotron2_model.py +++ b/tacotron_melgan/wrap_tacotron2_model.py @@ -11,7 +11,7 @@ from math import sqrt from tacotron_model import Tacotron2 -torch.set_default_tensor_type('torch.DoubleTensor') +#torch.set_default_tensor_type('torch.DoubleTensor') #torch.set_default_tensor_type('torch.FloatTensor') print('Loading tacotron model...') @@ -77,7 +77,7 @@ print('Load state dict...') module.load_state_dict(tacotron_model['state_dict']) #module.eval() # NB: Complains unless called: Did you forget call .eval() on your model? -output_filename = 'tacotron_container2.pt' +output_filename = 'tacotron_container3.pt' # NB: Tracing evaluates the model on input and unrolls and hardcodes branching # and loops. Scripting allows these to remain by converting the entire program