Update onnx_export.py
This commit is contained in:
parent
89a0de2834
commit
14861f6a05
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue