implement first GNO

2025-09-25 14:44:39 +02:00
parent d53b076ecc
commit f3be9e99f8
5 changed files with 89 additions and 306 deletions
--- a/ThermalSolver/data_module.py
+++ b/ThermalSolver/data_module.py
@@ -64,7 +64,7 @@ class GraphDataModule(LightningDataModule):
        edge_attr = torch.cat(
            [edge_attr, torch.norm(edge_attr, dim=1).unsqueeze(-1)], dim=1
        )
-        
+
        return Data(
            x=boundary_vales.unsqueeze(-1),
            c=conductivity.unsqueeze(-1),