Update onnx_export.py

SayaSS · Apr 18, 2023 · 5570594 · 5570594
1 parent 578c243
commit 5570594
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/onnx_export.py b/onnx_export.py
@@ -17,7 +17,9 @@ def main(NetExport):
             i.requires_grad = False
 
         n_frame = 10
-        test_hidden_unit = torch.rand(1, n_frame, 256)
+        hidden_channels = 256  #(Hubert's shape[2])
+
+        test_hidden_unit = torch.rand(1, n_frame, hidden_channels)
         test_pitch = torch.rand(1, n_frame)
         test_mel2ph = torch.arange(0, n_frame, dtype=torch.int64)[None] # torch.LongTensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]).unsqueeze(0)
         test_uv = torch.ones(1, n_frame, dtype=torch.float32)