Track and save only changed w and epsilon

2025-07-03 11:00:03 +02:00 · 2022-04-30 16:43:13 +02:00 · 2022-04-30 16:43:13 +02:00 · f3385b34ac
commit f3385b34ac
parent fba40c9b83
1 changed files with 35 additions and 33 deletions
--- a/learn_it.py
+++ b/learn_it.py
@ -452,6 +452,7 @@ with torch.no_grad():
                                eps_xy[id], dtype=torch.float64
                            )

+                        if cfg.network_structure.w_trainable[id] is True:
                            # Save the new values
                            np.save(
                                cfg.weight_path
@ -472,6 +473,7 @@ with torch.no_grad():
                            except ValueError:
                                pass

+                        if cfg.network_structure.eps_xy_trainable[id] is True:
                            np.save(
                                cfg.eps_xy_path
                                + "/EpsXY_L"