Update generation
Browse files- notes/generation.py +3 -3
notes/generation.py
CHANGED
@@ -40,7 +40,7 @@ target_column = "targets"
|
|
40 |
max_source_length = 256
|
41 |
max_target_length = 1024
|
42 |
seed = 42
|
43 |
-
eval_batch_size =
|
44 |
# generation_kwargs = {
|
45 |
# "max_length": 1024,
|
46 |
# "min_length": 128,
|
@@ -51,10 +51,10 @@ eval_batch_size = 32
|
|
51 |
# }
|
52 |
generation_kwargs = {
|
53 |
"max_length": 1024,
|
54 |
-
"min_length":
|
55 |
"no_repeat_ngram_size": 3,
|
56 |
"early_stopping": True,
|
57 |
-
"num_beams":
|
58 |
"length_penalty": 1.5,
|
59 |
}
|
60 |
|
|
|
40 |
max_source_length = 256
|
41 |
max_target_length = 1024
|
42 |
seed = 42
|
43 |
+
eval_batch_size = 64
|
44 |
# generation_kwargs = {
|
45 |
# "max_length": 1024,
|
46 |
# "min_length": 128,
|
|
|
51 |
# }
|
52 |
generation_kwargs = {
|
53 |
"max_length": 1024,
|
54 |
+
"min_length": 64,
|
55 |
"no_repeat_ngram_size": 3,
|
56 |
"early_stopping": True,
|
57 |
+
"num_beams": 4,
|
58 |
"length_penalty": 1.5,
|
59 |
}
|
60 |
|