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

Track and save only changed w and epsilon

Browse source
This commit is contained in:
David Rotermund 2022-04-30 16:43:13 +02:00 committed by GitHub
parent fba40c9b83
commit f3385b34ac
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 35 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
learn_it.py
View file

@ -452,6 +452,7 @@ with torch.no_grad():
eps_xy[id], dtype=torch.float64 eps_xy[id], dtype=torch.float64
) )
if cfg.network_structure.w_trainable[id] is True:
# Save the new values # Save the new values
np.save( np.save(
cfg.weight_path cfg.weight_path
@ -472,6 +473,7 @@ with torch.no_grad():
except ValueError: except ValueError:
pass pass
if cfg.network_structure.eps_xy_trainable[id] is True:
np.save( np.save(
cfg.eps_xy_path cfg.eps_xy_path
+ "/EpsXY_L" + "/EpsXY_L"