adjust wandb account info, lr scheduler for version issue

QData / spacetimeformer

Multivariate Time Series Forecasting with efficient Transformers. Code for the paper "Long-Range Transformers for Dynamic Spatiotemporal Forecasting."

https://arxiv.org/abs/2109.12218

MIT License

808 stars 191 forks source link

adjust wandb account info, lr scheduler for version issue #26

Closed jakegrigsby closed 2 years ago

jakegrigsby commented 2 years ago

Our pytorch lightning training loop was written for version 1.3, which has quickly led to some versioning issues with the current v1.5.8 - especially in logging and lr scheduling. This should fix #17 and #21.

Also makes it easier to customize wandb account information using environment variables instead of directly editing the train script.