luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gnn_4

Browse Source
This commit is contained in:
luni
2023-02-24 21:32:02 +00:00
parent 9a74c476a6
commit 53bfbabcd7
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/gnn.py
+3 -3
View File
@@ -16,9 +16,9 @@ class Smoother(tf.keras.Model):
def __init__(self, ): def __init__(self, ):
super().__init__() super().__init__()
self.a = sp_matrix_to_sp_tensor(sio.loadmat("custom_utils/ssmpl0.mat")['L'].tocoo()) self.a = sp_matrix_to_sp_tensor(sio.loadmat("custom_utils/ssmpl0.mat")['L'].tocoo())
self.conv1 = GCNConv(64, activation='tanh', use_bias=False) self.conv1 = GCNConv(3, activation='tanh', use_bias=False)
self.conv2 = GCNConv(64, activation='tanh', use_bias=False) self.conv2 = GCNConv(3, activation='tanh', use_bias=False)
self.conv3 = GCNConv(64, activation='tanh', use_bias=False) self.conv3 = GCNConv(3, activation='tanh', use_bias=False)
# self.conv4 = GCNConv(64, activation='tanh', use_bias=False) # self.conv4 = GCNConv(64, activation='tanh', use_bias=False)
# self.conv5 = GCNConv(64, activation='tanh', use_bias=False) # self.conv5 = GCNConv(64, activation='tanh', use_bias=False)
self.conv6 = GCNConv(3, activation='tanh', use_bias=False) self.conv6 = GCNConv(3, activation='tanh', use_bias=False)