Spaces:
Running
Running
Print run info
Browse files- 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.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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)
|