Update README.md
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ model_outputs.logits
|
|
35 |
|
36 |
# For generation. Pardon the messiness. Note the decoder_start_token_id.
|
37 |
|
38 |
-
model.eval() #
|
39 |
|
40 |
model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=20, min_length=1, early_stopping=True, pad_token_id=pad_id, bos_token_id=bos_id, eos_token_id=eos_id, decoder_start_token_id=tokenizer._convert_token_to_id_with_added_voc("<2en>"))
|
41 |
|
|
|
35 |
|
36 |
# For generation. Pardon the messiness. Note the decoder_start_token_id.
|
37 |
|
38 |
+
model.eval() # Set dropouts to zero
|
39 |
|
40 |
model_output=model.generate(inp, use_cache=True, num_beams=4, max_length=20, min_length=1, early_stopping=True, pad_token_id=pad_id, bos_token_id=bos_id, eos_token_id=eos_id, decoder_start_token_id=tokenizer._convert_token_to_id_with_added_voc("<2en>"))
|
41 |
|