diff --git a/vocoder/tf/convert_melgan_torch_to_tf.py b/vocoder/tf/convert_melgan_torch_to_tf.py index 1b3d687d..4c8515d9 100644 --- a/vocoder/tf/convert_melgan_torch_to_tf.py +++ b/vocoder/tf/convert_melgan_torch_to_tf.py @@ -100,13 +100,18 @@ for i in range(1, len(model.layers)): diff = compare_torch_tf(out_torch, out_tf_) assert diff < 1e-5, diff -dummy_input_torch = torch.ones((1, 80, 10)) +torch.manual_seed(0) +dummy_input_torch = torch.rand((1, 80, 100)) dummy_input_tf = tf.convert_to_tensor(dummy_input_torch.numpy()) +model.inference_padding = 0 +model_tf.inference_padding = 0 output_torch = model.inference(dummy_input_torch) output_tf = model_tf(dummy_input_tf, training=False) assert compare_torch_tf(output_torch, output_tf) < 1e-5, compare_torch_tf( output_torch, output_tf) + # save tf model save_checkpoint(model_tf, checkpoint['step'], checkpoint['epoch'], args.output_path) print(' > Model conversion is successfully completed :).') +