—1326上午去参观新时代展览啦~
下午先来看看gnn代码叭~一会还要去做工区
一、GNN@lab0
1、先复习了一下,马上做工区啦
2、pos=nx.spring_layer(G, seed=42)
—1358终于回来啦
3、pytorch中的Print(model)只能显示基本构件,不能显示forward中的model
使用torchinfo中的summary就好啦
参考链接
import torch
from torch.nn import Linear
from torch_geometric.nn import GCNConv
from torchinfo import summary
class GCN(torch.nn.Module):
def __init__(self,input_dim, hidden_dim,num_layers):
super(GCN, self).__init__()
torch.manual_seed(12345)
self.classifier = Linear(