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:
committed by
Nicola Demo
parent
30f865d912
commit
2be57944ba
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user