Safran/KPConv-PyTorch
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

relu at head softmax correction

Browse source
This commit is contained in:
HuguesTHOMAS 2021-07-30 20:46:01 +00:00
parent e13cc3874d
commit 3d683b6bd6
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
models/architectures.py
View file

@ -116,7 +116,7 @@ class KPCNN(nn.Module):
block_in_layer = 0
self.head_mlp = UnaryBlock(out_dim, 1024, False, 0)
self.head_softmax = UnaryBlock(1024, config.num_classes, False, 0)
self.head_softmax = UnaryBlock(1024, config.num_classes, False, 0, no_relu=True)
################
# Network Losses
@ -294,7 +294,7 @@ class KPFCNN(nn.Module):
out_dim = out_dim // 2
self.head_mlp = UnaryBlock(out_dim, config.first_features_dim, False, 0)
self.head_softmax = UnaryBlock(config.first_features_dim, self.C, False, 0)
self.head_softmax = UnaryBlock(config.first_features_dim, self.C, False, 0, no_relu=True)
################
# Network Losses

4
plot_convergence.py
View file

@ -707,7 +707,9 @@ def experiment_name_1():
logs = np.sort([join(res_path, l) for l in listdir(res_path) if start <= l <= end])
# Give names to the logs (for plot legends)
logs_names = ['name_log_1']
logs_names = ['name_log_1',
'name_log_2',
'name_log_3']
# safe check log names
logs_names = np.array(logs_names[:len(logs)])