Update onnx_export.py

2023-03-26 00:28:10 +08:00 · 2023-03-26 00:28:10 +08:00 · 14861f6a05
parent 89a0de2834
commit 14861f6a05
1 changed files with 1 additions and 1 deletions
--- a/onnx_export.py
+++ b/onnx_export.py
@ -16,7 +16,7 @@ def main(NetExport):
        
        test_hidden_unit = torch.rand(1, 10, 256) # rand
        test_pitch = torch.rand(1, 10) # rand
-        test_mel2ph = torch.LongTensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]).unsqueeze(0)
+        test_mel2ph = torch.arange(0, 10, dtype=torch.int64)[None]
        #test_mel2ph = torch.LongTensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19]).unsqueeze(0)
        test_uv = torch.zeros(1, 2048, 10, dtype=torch.float32)
        test_uv += 0.0064