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

Gradient accumulation in BPTT (#2)

Browse Source
This commit is contained in:
Filippo Olivo
2025-11-11 20:14:28 +01:00
committed by GitHub
parent 195c66b444
commit a2dd348423
4 changed files with 292 additions and 179 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ThermalSolver/model/__init__.py
View File

@@ -1,13 +1,13 @@
__all__ = [
"GraphFiniteDifference",
# "GraphFiniteDifference",
"GatingGNO",
"LearnableGraphFiniteDifference",
# "LearnableGraphFiniteDifference",
"PointNet",
]
from .learnable_finite_difference import (
GraphFiniteDifference as LearnableGraphFiniteDifference,
)
from .finite_difference import GraphFiniteDifference as GraphFiniteDifference
# from .learnable_finite_difference import (
# GraphFiniteDifference as LearnableGraphFiniteDifference,
# )
# from .finite_difference import GraphFiniteDifference as GraphFiniteDifference
from .local_gno import GatingGNO
from .point_net import PointNet