Closed xunil17 closed 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.
The hetero_gc_lstm model uses python dictionaries which causes errors when moving the model to different devices. Changed to utilize nn.ParameterDict.