More working than not

Not sure if validation works, call it a day
2024-03-03 03:16:54 +00:00 · 2024-03-03 03:16:54 +00:00 · 12aabb0d3f
commit 12aabb0d3f
parent 4a03211c83
10 changed files with 116 additions and 105 deletions
--- a/model/glue.py
+++ b/model/glue.py
@ -46,7 +46,7 @@ class SquareCropTransformLayer(nn.Module):
            torch.tensor_split(
                torch.cat(
                    torch.tensor_split(
-                        t_,
+                        kpoints_,
                        h_split_count,
                        dim=1
                    )