variable name fix FeedForward model

This commit is contained in:
Dario Coscia
2023-09-13 12:48:55 +02:00
committed by Nicola Demo
parent 17464ceca9
commit 5a4c114d48
3 changed files with 29 additions and 29 deletions

View File

@@ -28,16 +28,16 @@ class FeedForward(torch.nn.Module):
`inner_size` are not considered.
:param bool bias: If `True` the MLP will consider some bias.
"""
def __init__(self, input_dimensons, output_dimensions, inner_size=20,
def __init__(self, input_dimensions, output_dimensions, inner_size=20,
n_layers=2, func=nn.Tanh, layers=None, bias=True):
"""
"""
super().__init__()
if not isinstance(input_dimensons, int):
raise ValueError('input_dimensons expected to be int.')
self.input_dimension = input_dimensons
if not isinstance(input_dimensions, int):
raise ValueError('input_dimensions expected to be int.')
self.input_dimension = input_dimensions
if not isinstance(output_dimensions, int):
raise ValueError('output_dimensions expected to be int.')