Added epsilon xy mean back in

2022-05-01 02:11:59 +02:00 · 2022-05-01 02:11:59 +02:00 · c2cc2b4b1f
commit c2cc2b4b1f
parent 4979a99e09
1 changed files with 2 additions and 0 deletions
--- 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