Diffusion Model parameter size

Y-debug-sys / Diffusion-TS

[ICLR 2024] Official Implementation of "Diffusion-TS: Interpretable Diffusion for General Time Series Generation"

MIT License

186 stars 27 forks source link

Diffusion Model parameter size #17

Closed OmegaAsh closed 7 months ago

OmegaAsh commented 7 months ago

Hi guys, I am just using this for a project (will be sure to cite). When using the parameter config for the FMRI dataset, I am getting a parameter size of over 13M, but I think in the paper it was around 1.3M. Just wondering if this is an issue that you have encountered, or if it is something that is related to my setup. Thanks,

Y-debug-sys commented 7 months ago

Hi, this looks strange. Where did you get the data from? Is it the size of the .pt file?

OmegaAsh commented 7 months ago

Hi thanks for the quick reply, I just figured out it was due to my large sequence length, so no issues with the code. Thanks again and I will close this.

l402l commented 7 months ago

你好，请问计算model size的函数在哪呢，我计算sine的参数为235249，文中显示的为232177

Y-debug-sys commented 7 months ago

你好，由于对最初版本代码的调整和可能存在的环境差异，导致模型大小的细微差别很正常。