def __init__(self, input_dim, output_dim, hidden_layer=64, activation='sin'):
    super(FFNN, self).__init__()
    self.activation = get_activation(activation)
    self.fc1 = LinearBlock(input_dim, hidden_layer)
    self.fc2 = LinearBlock(hidden_layer, hidden_layer)
    self.fc3 = LinearBlock(hidden_layer, hidden_layer)
    self.fc4 = LinearBlock(hidden_layer, hidden_layer)
    self.fc5 = LinearBlock(hidden_layer, output_dim)
def __init__(self, input_dim, output_dim, hidden_layer=64, activation='sin'):
    super(FFNN, self).__init__()
    self.activation = get_activation(activation)
    self.fc1 = LinearBlock(input_dim, hidden_layer)
    self.fc2 = LinearBlock(hidden_layer, hidden_layer)
    self.fc3 = LinearBlock(hidden_layer, hidden_layer)
    self.fc4 = LinearBlock(hidden_layer, hidden_layer)
    self.fc5 = LinearBlock(hidden_layer, output_dim)