Fix SupervisedSolver GPU bug and implement GraphSolver (#346)

* Fix some bugs
* Solve bug with GPU and model_summary parameters in SupervisedSolver class
* Implement GraphSolver class
* Fix Tutorial 5
This commit is contained in:
FilippoOlivo
2024-09-21 18:55:57 +02:00
committed by Nicola Demo
parent 30f865d912
commit 2be57944ba
10 changed files with 334 additions and 164 deletions

View File

@@ -269,4 +269,7 @@ class FNO(KernelNeuralOperator):
:return: The output tensor obtained from FNO.
:rtype: torch.Tensor
"""
if isinstance(x, LabelTensor):
x = x.as_subclass(torch.Tensor)
return super().forward(x)