Iker commited on
Commit
f85b100
·
1 Parent(s): 122365e

Print run info

Browse files
Files changed (1) hide show
  1. translate.py +10 -1
translate.py CHANGED
@@ -101,7 +101,16 @@ def main(
101
 
102
  total_lines: int = count_lines(sentences_path)
103
  print(
104
- f"We will translate {total_lines} lines. Initial batch size: {starting_batch_size}"
 
 
 
 
 
 
 
 
 
105
  )
106
 
107
  @find_executable_batch_size(starting_batch_size=starting_batch_size)
 
101
 
102
  total_lines: int = count_lines(sentences_path)
103
  print(
104
+ f"** We will translate {total_lines} lines. **\n"
105
+ f"Input file: {sentences_path}\n"
106
+ f"Output file: {output_path}\n"
107
+ f"Source language: {source_lang}\n"
108
+ f"Target language: {target_lang}\n"
109
+ f"Starting batch size: {starting_batch_size}\n"
110
+ f"Max length: {max_length}\n"
111
+ f"Num beams: {num_beams}\n"
112
+ f"Precision: {model.dtype}\n"
113
+ f"Model: {model_name}\n"
114
  )
115
 
116
  @find_executable_batch_size(starting_batch_size=starting_batch_size)