Change dicts in hetero_gc_lstm to nn.ParameterDicts

benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

MIT License

2.61k stars 367 forks source link

Change dicts in hetero_gc_lstm to nn.ParameterDicts #180

Closed xunil17 closed 2 years ago

xunil17 commented 2 years ago

The hetero_gc_lstm model uses python dictionaries which causes errors when moving the model to different devices. Changed to utilize nn.ParameterDict.