Skip to content

GitLab

"tests/python/common/test_heterograph-kernel.py" did not exist on "dd65ee211ea5ec1c876d323c4387f066bee41a77"
Unverified Commit a64ff482 authored by Andrei Ivanov's avatar Andrei Ivanov Committed by GitHub
Browse files

Removing the UserWarning appearing in the `mnist` example. (#6000) 


Co-authored-by: default avatarHongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
parent 83115794
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/pytorch/model_zoo/geometric/mnist.py
View file @ a64ff482
...@@ -109,7 +109,7 @@ class MoNet(nn.Module): ...@@ -109,7 +109,7 @@ class MoNet(nn.Module):
) )
self.cls = nn.Sequential( self.cls = nn.Sequential(
nn.Linear(hiddens[-1], out_feats), nn.LogSoftmax() nn.Linear(hiddens[-1], out_feats), nn.LogSoftmax(dim=1)
) )
def forward(self, g_arr, feat): def forward(self, g_arr, feat):
...@@ -137,7 +137,7 @@ class ChebNet(nn.Module): ...@@ -137,7 +137,7 @@ class ChebNet(nn.Module):
self.layers.append(ChebConv(hiddens[i - 1], hiddens[i], k)) self.layers.append(ChebConv(hiddens[i - 1], hiddens[i], k))
self.cls = nn.Sequential( self.cls = nn.Sequential(
nn.Linear(hiddens[-1], out_feats), nn.LogSoftmax() nn.Linear(hiddens[-1], out_feats), nn.LogSoftmax(dim=1)
) )
def forward(self, g_arr, feat): def forward(self, g_arr, feat):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment