From c2cc2b4b1fbdd004f930f5c31639697936e6fff4 Mon Sep 17 00:00:00 2001
From: David Rotermund <54365609+davrot@users.noreply.github.com>
Date: Sun, 1 May 2022 02:11:59 +0200
Subject: [PATCH] Added epsilon xy mean back in

---
 learn_it.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/learn_it.py b/learn_it.py
index b01d345..9431de3 100644
--- a/learn_it.py
+++ b/learn_it.py
@@ -465,6 +465,8 @@ with torch.no_grad():
                             network[id].threshold_epsilon_xy(
                                 cfg.learning_parameters.learning_rate_threshold_eps_xy
                             )
+                            if cfg.network_structure.eps_xy_mean[id] is True:
+                                network[id].mean_epsilon_xy()
                         else:
                             network[id].epsilon_xy = torch.tensor(
                                 eps_xy[id], dtype=torch.float64