From 74d618774a1900ead102da4ff309660d6ffa4f21 Mon Sep 17 00:00:00 2001 From: steffen-illium Date: Sun, 23 May 2021 10:36:22 +0200 Subject: [PATCH] application losses II --- journal_robustness.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/journal_robustness.py b/journal_robustness.py index 96c7942..0fbb296 100644 --- a/journal_robustness.py +++ b/journal_robustness.py @@ -105,10 +105,9 @@ class RobustnessComparisonExperiment: for i, fixpoint in enumerate(self.id_functions): row_headers.append(fixpoint.name) + loss_per_application = [[0 for _ in range(10)] for _ in range(len(self.id_functions))] for seed in range(10): for noise_level in noise_range: - application_losses = [] - clone = Net(fixpoint.input_size, fixpoint.hidden_size, fixpoint.out_size, f"{fixpoint.name}_clone_noise10e-{noise_level}") clone.load_state_dict(copy.deepcopy(fixpoint.state_dict())) @@ -129,7 +128,8 @@ class RobustnessComparisonExperiment: clone_weight_post_application = clone.input_weight_matrix() target_data_post_application = clone.create_target_weights(clone_weight_post_application) - application_losses.append(F.l1_loss(target_data_pre_application, target_data_post_application)) + loss_per_application[seed][noise_level] = (F.l1_loss(target_data_pre_application, + target_data_post_application)) if print_it: