search

microsoft / ProbTS

ProbTS is a benchmarking toolkit for time series forecasting.
MIT License
88 stars 10 forks source link

UniTS model hard-coded "cuda:0" when loading pretrain weights #23

Open zhangzw16 opened 2 months ago

zhangzw16 commented 2 months ago

https://github.com/microsoft/ProbTS/blob/d299e1cc26df337b9aff9e5951df7d652f74c2f5/probts/model/forecaster/point_forecaster/units.py#L1014

This should be made more robust to support various devices.

zhangzw16 commented 2 months ago

Fixed with #24

zhangzw16 commented 2 months ago

Maybe need a better way to specify device when inference.