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

big gnn

Browse Source
This commit is contained in:
luni
2023-02-27 08:52:35 +00:00
parent c294f1f0b7
commit f479643be6
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccv/config_abl_all.py
+2 -2
View File
@@ -19,8 +19,8 @@ config['only_curvatures'] = False
ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False) ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False)
assert ablation_count == 1 or ablation_count == 0 assert ablation_count == 1 or ablation_count == 0
config['gpu'] = '2' config['gpu'] = '0'
config['name'] = 'abl_all_gnn' config['name'] = 'abl_all_gnn_2'
config['notes'] = '' config['notes'] = ''
config['tags'] = ['iccv', 'abl_all_gnn'] config['tags'] = ['iccv', 'abl_all_gnn']
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(16, activation='tanh', use_bias=False) self.conv1 = GCNConv(512, activation='tanh', use_bias=False)
self.conv2 = GCNConv(16, activation='tanh', use_bias=False) self.conv2 = GCNConv(512, activation='tanh', use_bias=False)
self.conv3 = GCNConv(16, activation='tanh', use_bias=False) self.conv3 = GCNConv(512, activation='tanh', use_bias=False)
# self.conv4 = GCNConv(16, activation='tanh', use_bias=False) self.conv4 = GCNConv(512, activation='tanh', use_bias=False)
# self.conv5 = GCNConv(16, 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