not bad this setup

2025-12-18 09:30:21 +01:00
parent 4fdf817d75
commit 0a034225ef
4 changed files with 120 additions and 58 deletions
--- a/ThermalSolver/graph_datamodule.py
+++ b/ThermalSolver/graph_datamodule.py
@@ -82,7 +82,7 @@ class GraphDataModule(LightningDataModule):
        conductivity = torch.tensor(
            snapshot["conductivity"], dtype=torch.float32
        )
-        temperature = torch.tensor(snapshot["temperature"], dtype=torch.float32)
+        temperature = torch.tensor(snapshot["temperature"], dtype=torch.float32)[:50]

        pos = torch.tensor(geometry["points"], dtype=torch.float32)[:, :2]