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

remove inner_size (#44)

Browse Source
This commit is contained in:
Francesco Andreuzzi
2022-12-05 10:50:28 +01:00
committed by GitHub
parent b6c88bb756
commit 399c72fc0e
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pina/model/deeponet.py
View File

@@ -18,7 +18,7 @@ class DeepONet(torch.nn.Module):
<https://doi.org/10.1038/s42256-021-00302-5>`_
"""
def __init__(self, branch_net, trunk_net, output_variables, inner_size=10):
def __init__(self, branch_net, trunk_net, output_variables):
"""
:param torch.nn.Module branch_net: the neural network to use as branch
model. It has to take as input a :class:`LabelTensor`. The number
@@ -29,8 +29,6 @@ class DeepONet(torch.nn.Module):
:param list(str) output_variables: the list containing the labels
corresponding to the components of the output computed by the
model.
:param int inner_size: the output dimension of the branch and trunk
networks. Default is 10.
:Example:
>>> branch = FFN(input_variables=['a', 'c'], output_variables=20)