folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🎨 Format Python code with psf/black

Browse Source
This commit is contained in:
ndem0
2024-02-09 11:25:00 +00:00
committed by Nicola Demo
parent 591aeeb02b
commit cbb43a5392
64 changed files with 1323 additions and 955 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pina/model/multi_feed_forward.py
View File

@@ -1,4 +1,5 @@
"""Module for Multi FeedForward model"""
import torch
from .feed_forward import FeedForward
@@ -6,7 +7,7 @@ from .feed_forward import FeedForward
class MultiFeedForward(torch.nn.Module):
"""
The PINA implementation of MultiFeedForward network.
The PINA implementation of MultiFeedForward network.
This model allows to create a network with multiple FeedForward combined
together. The user has to define the `forward` method choosing how to