search

TorchSpatiotemporal / tsl

tsl: a PyTorch library for processing spatiotemporal data.
https://torch-spatiotemporal.readthedocs.io/
MIT License
236 stars 22 forks source link

add support for current torchvision & tsl versions #36

Closed Abusagit closed 2 months ago

Abusagit commented 2 months ago

Main file for training a model in the forecasting experiment contained deprecated argument for a metric, as well as DCRNN contained parameters which aren't presented in its signature in current version.

marshka commented 2 months ago

PR merged, thanks for spotting this!

Abusagit commented 2 months ago

Thank you! If you don't mind, I can do it for other models which I'll launch as well.

By the way, this is very thorough training template for training a model and I like that you used Hydra here, such a powerful tool! :)

marshka commented 2 months ago

Sure, contributions are more than welcome! Glad that you liked our example :)