Skip to content

Commit df9d88c

Browse files
author
zysite
committed
Set default dropout ratio to 0
1 parent c38617b commit df9d88c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

parser/modules/mlp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
class MLP(nn.Module):
99

10-
def __init__(self, n_in, n_hidden, dropout):
10+
def __init__(self, n_in, n_hidden, dropout=0):
1111
super(MLP, self).__init__()
1212

1313
self.linear = nn.Linear(n_in, n_hidden)

0 commit comments

Comments
 (0)