pszemraj commited on
Commit
aa3c57c
1 Parent(s): af3c92a

Cast hyperparameter types

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -41,11 +41,11 @@ def proc_submission(
41
  """
42
 
43
  settings = {
44
- "length_penalty": length_penalty,
45
- "repetition_penalty": repetition_penalty,
46
- "no_repeat_ngram_size": no_repeat_ngram_size,
47
  "encoder_no_repeat_ngram_size": 4,
48
- "num_beams": num_beams,
49
  "min_length": 4,
50
  "max_length": int(token_batch_length // 4),
51
  "early_stopping": True,
 
41
  """
42
 
43
  settings = {
44
+ "length_penalty": float(length_penalty),
45
+ "repetition_penalty": float(repetition_penalty),
46
+ "no_repeat_ngram_size": int(no_repeat_ngram_size),
47
  "encoder_no_repeat_ngram_size": 4,
48
+ "num_beams": int(num_beams),
49
  "min_length": 4,
50
  "max_length": int(token_batch_length // 4),
51
  "early_stopping": True,