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

Compare commits

base: luni/cvpr:68b2d0d3affe9b4c587c79cc64700cec6dbb1c40
Branches Tags
luni/cvpr:master luni/cvpr:iccv_abl luni/cvpr:seq_pyr_fine luni/cvpr:iccv_abl_fine luni/cvpr:seq_pyr luni/cvpr:iccv_resnet luni/cvpr:pretrained_cape luni/cvpr:iccv_abl_gnn luni/cvpr:gnn_frozen luni/cvpr:iccv luni/cvpr:gnn luni/cvpr:gnn_decoder luni/cvpr:new_model luni/cvpr:review_2_vis luni/cvpr:review_3_orig luni/cvpr:review_4_orig luni/cvpr:review_2_orig luni/cvpr:review_5_orig luni/cvpr:review_1_orig luni/cvpr:review_2_raw luni/cvpr:review_7_orig luni/cvpr:review_7 luni/cvpr:post_orig luni/cvpr:review_1
...
compare: luni/cvpr:53bfbabcd73e01084790718105f4734e03eb3253
Branches Tags
luni/cvpr:iccv_abl luni/cvpr:seq_pyr_fine luni/cvpr:iccv_abl_fine luni/cvpr:seq_pyr luni/cvpr:iccv_resnet luni/cvpr:pretrained_cape luni/cvpr:iccv_abl_gnn luni/cvpr:gnn_frozen luni/cvpr:iccv luni/cvpr:gnn luni/cvpr:gnn_decoder luni/cvpr:new_model luni/cvpr:master luni/cvpr:review_2_vis luni/cvpr:review_3_orig luni/cvpr:review_4_orig luni/cvpr:review_2_orig luni/cvpr:review_5_orig luni/cvpr:review_1_orig luni/cvpr:review_2_raw luni/cvpr:review_7_orig luni/cvpr:review_7 luni/cvpr:post_orig luni/cvpr:review_1
2 Commits
68b2d0d3af ... 53bfbabcd7
Author SHA1 Message Date
luni 53bfbabcd7 gnn_4 2023-02-24 21:32:02 +00:00
luni 9a74c476a6 gnn_3 2023-02-24 21:31:37 +00:00
1 changed files with 7 additions and 7 deletions
Expand all files Collapse all files
models/gnn.py
+7 -7
View File
@@ -16,11 +16,11 @@ 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)
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