full compatibility with torch models

* Network class added
* adding tests for Network class
This commit is contained in:
Dario Coscia
2022-11-08 12:11:58 +01:00
committed by GitHub
parent a92a764844
commit bb1efe44bc
3 changed files with 189 additions and 0 deletions

View File

@@ -2,8 +2,10 @@ __all__ = [
'FeedForward',
'MultiFeedForward'
'DeepONet',
'Network'
]
from .feed_forward import FeedForward
from .multi_feed_forward import MultiFeedForward
from .deeponet import DeepONet
from .network import Network