propagate the dropout flag
This commit is contained in:
@@ -142,7 +142,7 @@ model_args = dict(
|
|||||||
vocab_size=32000,
|
vocab_size=32000,
|
||||||
multiple_of=multiple_of,
|
multiple_of=multiple_of,
|
||||||
max_seq_len=max_seq_len,
|
max_seq_len=max_seq_len,
|
||||||
#dropout=dropout,
|
dropout=dropout,
|
||||||
) # start with model_args from command line
|
) # start with model_args from command line
|
||||||
if init_from == "scratch":
|
if init_from == "scratch":
|
||||||
# init a new model from scratch
|
# init a new model from scratch
|
||||||
|
|||||||
Reference in New Issue
Block a user