Skip to content

Commit

Permalink
Update docstring for multiple output heads
Browse files Browse the repository at this point in the history
  • Loading branch information
akashvelu committed Jun 8, 2021
1 parent cf90ff6 commit f25baa9
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion nupic/research/frameworks/dendrites/modules/dendritic_mlp.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,9 @@ class DendriticMLP(nn.Module):
initializations and learning parameters
:param input_size: size of the input to the network
:param output_size: the number of units in the output layer
:param output_size: the number of units in the output layer. Must be either an
integer if there is a single output head, or an iterable
of integers if there are multiple output heads.
:param hidden_sizes: the number of units in each hidden layer
:param num_segments: the number of dendritic segments that each hidden unit has
:param dim_context: the size of the context input to the network
Expand Down

0 comments on commit f25baa9

Please sign in to comment.