BatchNorm1d (dim) nn2 = Sequential(Linear(dim, dim), ReLU(), Linear(dim, dim)) self.conv2 = GINLAFConv(nn2, units=units, node_dim=dim) self.bn2 = torch.nn.
確定! 回上一頁