diff --git a/model/DDGCRN/DDGCRN.py b/model/DDGCRN/DDGCRN.py index cdf1cc9..368b3dc 100644 --- a/model/DDGCRN/DDGCRN.py +++ b/model/DDGCRN/DDGCRN.py @@ -116,3 +116,4 @@ class DGCN(nn.Module): D_inv = torch.diag_embed(torch.sum(graph, -1) ** (-0.5)) return torch.matmul(torch.matmul(D_inv, graph), D_inv) if normalize else torch.matmul( torch.matmul(D_inv, graph + I), D_inv) + \ No newline at end of file