Skip to content

Commit 3c52230

Browse files
authored
Update train.py
1 parent cbaacee commit 3c52230

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

DrQA/train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,8 @@
6161
help='perform rnn padding (much slower but more accurate).')
6262
# model
6363
parser.add_argument('--question_merge', default='self_attn')
64-
parser.add_argument('--doc_layers', type=int, default=3)
65-
parser.add_argument('--question_layers', type=int, default=3)
64+
parser.add_argument('--doc_layers', type=int, default=5)
65+
parser.add_argument('--question_layers', type=int, default=5)
6666
parser.add_argument('--hidden_size', type=int, default=128)
6767
parser.add_argument('--num_features', type=int, default=4)
6868
parser.add_argument('--pos', type=str2bool, nargs='?', const=True, default=True,

0 commit comments

Comments
 (0)