add plotting and checkpoints

2025-12-05 20:56:12 +01:00 · 2025-12-05 20:56:12 +01:00 · 65515d6d90
commit 65515d6d90
parent 6997ca3551
2 changed files with 65 additions and 2 deletions
--- a/flake.nix
+++ b/flake.nix
@ -101,6 +101,7 @@
            (pkgs.python3.withPackages (ppkgs: [
              ppkgs.torch
              ppkgs.tqdm
+              ppkgs.matplotlib
              (lib.python_package ppkgs)
            ]))
          ];