Results 17 comments of Elchin

Hi, I have problems with training this. Model always predicts "mean" spectrogram. I think, that problem is soft-dtw. Have you any idea how I can fix that? Or how can...

@hhguo does you have any solution for that?

I have also problem with duration predictor and residual encoder. Duration predictor mostly predicts one value for every symbol from input with small variation (all values in range 4-6). Besides...

I paste my piece of code here: https://pastebin.com/GFyguNP2 . Maybe it can help someone. Or you find some kind of bug.. P.S. you can also draw `D_xy.grad` to see aligner...

Of course, It works with onnxruntime and polygraphy. Polygraphy output ``` [I] RUNNING | Command: /home/user/conda/envs/ekerimov-convert/bin/polygraphy run onnx_500k/generator.onnx --onnxrt [I] onnxrt-runner-N0-01/29/24-15:45:19 | Activating and starting inference [I] Creating ONNX-Runtime Inference...

I couldn't find wheel in tar package of current repo. But I found in such archives https://developer.nvidia.com/nvidia-tensorrt-8x-download , but there is also version 8.6.1 I uploaded onnx model to reproduce...

And use such code ```python # %% import tensorrt as trt import onnx logger = trt.Logger(trt.Logger.VERBOSE) builder = trt.Builder(logger) network = builder.create_network(1

I found that the error is due to this line https://github.com/jaywalnut310/vits/blob/main/models.py#L517. or rather because of `attn.squeeze()`. But due squeeze doesn't work https://github.com/NVIDIA/TensorRT/issues/2846 I used just `attn = attn[:, 0]` and...