davrot/pytorch-sbs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added epsilon xy mean back in

Browse source
This commit is contained in:
David Rotermund 2022-05-01 02:11:59 +02:00 committed by GitHub
parent 4979a99e09
commit c2cc2b4b1f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
learn_it.py
View file

@ -465,6 +465,8 @@ with torch.no_grad():
network[id].threshold_epsilon_xy( network[id].threshold_epsilon_xy(
cfg.learning_parameters.learning_rate_threshold_eps_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: else:
network[id].epsilon_xy = torch.tensor( network[id].epsilon_xy = torch.tensor(
eps_xy[id], dtype=torch.float64 eps_xy[id], dtype=torch.float64