fix examples (#21)

This commit is contained in:
Nicola Demo
2022-07-21 13:41:59 +02:00
committed by GitHub
parent 62f203fcc3
commit e8c2f87460
14 changed files with 63 additions and 67 deletions

View File

@@ -7,7 +7,7 @@ from torch.nn import ReLU, Tanh, Softplus
from pina import PINN, LabelTensor, Plotter
from pina.model import FeedForward
from pina.adaptive_functions import AdaptiveSin, AdaptiveCos, AdaptiveTanh
from poisson2 import Poisson
from problems.poisson import Poisson
class myFeature(torch.nn.Module):
@@ -61,6 +61,4 @@ if __name__ == "__main__":
else:
pinn.load_state('pina.poisson')
plotter = Plotter()
plotter.plot(pinn, component='u')
plotter.plot(pinn)