folivo/thermal-conduction-ml
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

new data format

Browse Source
This commit is contained in:
Filippo Olivo
2025-12-12 10:18:16 +01:00
parent 27c2aeb736
commit 732d48c360
3 changed files with 118 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ThermalSolver/model/diffusion_net.py
View File

@@ -33,14 +33,12 @@ class DiffusionLayer(MessagePassing):
@property
def alpha(self):
return torch.clamp(self.alpha_param, min=1e-5, max=1.0)
return torch.clamp(self.alpha_param, min=1e-7, max=1.0)
def forward(self, x, edge_index, edge_weight, conductivity):
edge_weight = edge_weight.unsqueeze(-1)
conductance = self.phys_encoder(edge_weight)
net_flux = self.propagate(edge_index, x=x, conductance=conductance)
# return (1-self.alpha) * x + self.alpha * net_flux
# return net_flux + x
return x + self.alpha * net_flux
def message(self, x_i, x_j, conductance):