Skip to content

Commit

Permalink
update comments
Browse files Browse the repository at this point in the history
  • Loading branch information
zshiqiang committed Oct 10, 2023
1 parent 85aead1 commit 18ffcc5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/omlt/neuralnet/layer.py
Original file line number Diff line number Diff line change
Expand Up @@ -294,11 +294,11 @@ def __init__(
)
if input_size[-1] % N != 0:
raise ValueError(
"Input size must equal to the number of nodes multiply the number of input node features"
"Input size must equal to the number of nodes multiplied by the number of input node features"
)
if output_size[-1] % N != 0:
raise ValueError(
"Output size must equal to the number of nodes multiply the number of output node features"
"Output size must equal to the number of nodes multiplied by the number of output node features"
)
self.__N = N
self.__gnn_input_size = input_size[-1] // N
Expand Down

0 comments on commit 18ffcc5

Please sign in to comment.