search

chingyaoc / ggnn.pytorch

A PyTorch Implementation of Gated Graph Sequence Neural Networks (GGNN)
MIT License
460 stars 92 forks source link

Removing these code and then it become origin GNN? #6

Closed guotong1988 closed 5 years ago

guotong1988 commented 5 years ago

https://github.com/JamesChuanggg/ggnn.pytorch/blob/master/model.py#L29-L40

        self.reset_gate = nn.Sequential(
            nn.Linear(state_dim*3, state_dim),
            nn.Sigmoid()
        )
        self.update_gate = nn.Sequential(
            nn.Linear(state_dim*3, state_dim),
            nn.Sigmoid()
        )
        self.tansform = nn.Sequential(
            nn.Linear(state_dim*3, state_dim),
            nn.Tanh()
        )

@JamesChuanggg Thank you very much!

guotong1988 commented 5 years ago

image 《Learning Convolutional Neural Networks for Graphs》