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

gnn with weights

Browse Source
This commit is contained in:
luni
2023-02-25 12:33:18 +00:00
parent 5062ed95d8
commit d8337bfcfe
4 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/gnn.py
+2 -2
View File
@@ -18,7 +18,7 @@ class Smoother(tf.keras.Model):
self.a = sp_matrix_to_sp_tensor(sio.loadmat("custom_utils/ssmpl0.mat")['L'].tocoo())
self.conv1 = GCNConv(features, activation=activation_middle, use_bias=False)
self.conv2 = GCNConv(features, activation=activation_middle, use_bias=False)
self.conv3 = GCNConv(features, activation=activation_middle, use_bias=False)
# self.conv3 = GCNConv(features, activation=activation_middle, use_bias=False)
# self.conv4 = GCNConv(64, activation='tanh', use_bias=False)
# self.conv5 = GCNConv(64, activation='tanh', use_bias=False)
self.conv6 = GCNConv(3 if not learn_weights else 24, activation=activation_final, use_bias=False)
@@ -27,7 +27,7 @@ class Smoother(tf.keras.Model):
x = inputs
x = self.conv1([x, self.a], training=training)
x = self.conv2([x, self.a], training=training)
x = self.conv3([x, self.a], training=training)
# x = self.conv3([x, self.a], training=training)
# x = self.conv4([x, self.a], training=training)
# x = self.conv5([x, self.a], training=training)
x = self.conv6([x, self.a], training=training)