partially fix documentation (#80)

This commit is contained in:
Anna Ivagnes
2023-04-18 10:49:01 +02:00
committed by GitHub
parent c536f8267f
commit da33aeae3a
11 changed files with 222 additions and 4 deletions

View File

@@ -1,13 +1,16 @@
"""Module for Multi FeedForward model"""
import torch
from .feed_forward import FeedForward
class MultiFeedForward(torch.nn.Module):
"""
:param dict dff_dict: dictionary of FeedForward networks.
"""
def __init__(self, dff_dict):
'''
'''
"""
"""
super().__init__()
if not isinstance(dff_dict, dict):