search

salesforce / CoST

PyTorch code for CoST: Contrastive Learning of Disentangled Seasonal-Trend Representations for Time Series Forecasting (ICLR 2022)
BSD 3-Clause "New" or "Revised" License
215 stars 43 forks source link

'--kernels' problem #4

Closed msonah closed 2 years ago

msonah commented 2 years ago

Thank for sharing the code. You set the parameter 'kernels' (Kernel sizes for mixture of AR experts module) in the code to 'None', when I run the code, it will report an error (Rearrange can't be applied to an empty list), please ask' How should kernels' be set up, may you give me some advice? Looking forward to your reply！

gorold commented 2 years ago

Thanks for the interest in our work! the kernels parameter can be set to a list of ints. Pleases check the scripts folder on how to use this parameter.