From 3d683b6bd6bf058135d3f9f155cd41595dc81c16 Mon Sep 17 00:00:00 2001 From: HuguesTHOMAS Date: Fri, 30 Jul 2021 20:46:01 +0000 Subject: [PATCH] relu at head softmax correction --- models/architectures.py | 4 ++-- plot_convergence.py | 4 +++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/models/architectures.py b/models/architectures.py index e8cb254..4442579 100644 --- a/models/architectures.py +++ b/models/architectures.py @@ -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 diff --git a/plot_convergence.py b/plot_convergence.py index e398106..9de6ee6 100644 --- a/plot_convergence.py +++ b/plot_convergence.py @@ -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)])