Peter commited on
Commit
b9e8529
1 Parent(s): 4aad54f

add max_length and default beams

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -68,8 +68,9 @@ def proc_submission(
68
  "no_repeat_ngram_size": no_repeat_ngram_size,
69
  "encoder_no_repeat_ngram_size": 4,
70
  "num_beams": num_beams,
 
 
71
  "early_stopping": True,
72
- "min_length": 10,
73
  "do_sample": False,
74
  }
75
 
@@ -130,7 +131,7 @@ def load_examples(examples_dir="examples"):
130
  for example in examples:
131
  with open(example, "r") as f:
132
  text = f.read()
133
- text_examples.append([text, "large", 1, 512, 0.7, 3.5, 3])
134
 
135
  return text_examples
136
 
 
68
  "no_repeat_ngram_size": no_repeat_ngram_size,
69
  "encoder_no_repeat_ngram_size": 4,
70
  "num_beams": num_beams,
71
+ "min_length": 4,
72
+ "max_length": int(token_batch_length // 4),
73
  "early_stopping": True,
 
74
  "do_sample": False,
75
  }
76
 
 
131
  for example in examples:
132
  with open(example, "r") as f:
133
  text = f.read()
134
+ text_examples.append([text, "large", 2, 512, 0.7, 3.5, 3])
135
 
136
  return text_examples
137