Some ideas behind the conversion from TensorFlow to pytorch:
- We replace
Sequentialof Tensorflow bynn.Modulein pytorch, where the layers are coordinated inforward - We replace Spektral's
GCNConvwithtorch_geometric'sGCNConvfor graphs - We replace TensorFlow's
GlobalAveragePooling1Dbytorch.meanover the time dimension
Some other logics have been updated (to document)