small fix

2025-10-05 10:36:23 +02:00
parent 7a6fbdb89c
commit 469b1c6e13
3 changed files with 142 additions and 40 deletions
--- a/ThermalSolver/module.py
+++ b/ThermalSolver/module.py
@@ -109,7 +109,6 @@ class GraphSolver(LightningModule):
            edge_index=edge_index,
            edge_attr=edge_attr,
            unrolling_steps=self.unrolling_steps,
-            plot_results=True,
            batch=batch.batch,
            pos=batch.pos,
        )