use assign argument if available in nn.Module.load_state_dict

Nixtla / neuralforecast

Scalable and user friendly neural :brain: forecasting algorithms.

https://nixtlaverse.nixtla.io/neuralforecast

Apache License 2.0

2.69k stars 312 forks source link

Closed jmoralez closed 3 days ago

jmoralez commented 3 weeks ago

Fixes #986

review-notebook-app[bot] commented 3 weeks ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.