search

hazdzz / STGCN

The PyTorch implementation of STGCN.
GNU Lesser General Public License v2.1
484 stars 106 forks source link

intention of "class Align(nn.Module) " #2

Closed jacobchen1 closed 3 years ago

jacobchen1 commented 3 years ago

Hey Hazdzz, thanks for your awesome work. I wonder why this "class Align(nn.Module):" has been applied into the module. What is the intention to use this class?

hazdzz commented 3 years ago

The purpose of class Align is aligning the input channel numbers and the output channel numbers. Although it is not mentioned in the paper, the author write it in his code. If you compare the code between author's and mine, you would find that I just abstract those three if conditional sentences from Gated Temporal Convolution Layer and Graph Convolution Layer to a class Align. That's it. As for why need to align the channel numbers, I strongly recommend you to ask author himself.