add docs

2023-04-18 10:49:57 +02:00
parent da33aeae3a
commit 736c78fd64
17 changed files with 292 additions and 172 deletions
--- a/pina/model/multi_feed_forward.py
+++ b/pina/model/multi_feed_forward.py
@@ -9,8 +9,9 @@ class MultiFeedForward(torch.nn.Module):
    :param dict dff_dict: dictionary of FeedForward networks.
    """
    def __init__(self, dff_dict):
-        """
-        """
+        '''
+        dff_dict: dict of FeedForward objects
+        '''
        super().__init__()

        if not isinstance(dff_dict, dict):