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

gnn_3

Browse Source
This commit is contained in:
luni
2023-02-24 21:31:37 +00:00
parent 68b2d0d3af
commit 9a74c476a6
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/gnn.py
+4 -4
View File
@@ -19,8 +19,8 @@ class Smoother(tf.keras.Model):
self.conv1 = GCNConv(64, activation='tanh', use_bias=False) self.conv1 = GCNConv(64, activation='tanh', use_bias=False)
self.conv2 = GCNConv(64, activation='tanh', use_bias=False) self.conv2 = GCNConv(64, activation='tanh', use_bias=False)
self.conv3 = GCNConv(64, activation='tanh', use_bias=False) self.conv3 = GCNConv(64, 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)
def call(self, inputs, training=None): def call(self, inputs, training=None):
@@ -28,8 +28,8 @@ class Smoother(tf.keras.Model):
x = self.conv1([x, self.a], training=training) x = self.conv1([x, self.a], training=training)
x = self.conv2([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.conv4([x, self.a], training=training)
x = self.conv5([x, self.a], training=training) # x = self.conv5([x, self.a], training=training)
x = self.conv6([x, self.a], training=training) x = self.conv6([x, self.a], training=training)
return x return x