search

zalandoresearch / pytorch-ts

PyTorch based Probabilistic Time Series forecasting framework based on GluonTS backend
MIT License
1.21k stars 191 forks source link

The purpose of the self.distr_args(rnn_outputs) #104

Closed zf223669 closed 1 year ago

zf223669 commented 1 year ago

Hi, I could not understand the code "distr_args = self.distr_args(rnn_outputs=rnn_outputs)" in time_grad_network.py, Could you explain more detail about that, what`s that purpose? Thank you :)

kashif commented 1 year ago

I believe the purpose of this was to project the output from the RNN to the conditioning vector length which the conditional denoising diffusion expects... however if the conditional denoising diffusion module expects the vector size to the same as the hidden size of the RNN then yes this part is perhaps not needed, but with this i can control the conditional vector size... hopefully this helps?

zf223669 commented 1 year ago

Kashif, thank you for your answer ：）