Spaces:
Runtime error
Runtime error
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)
|