Spaces:
Running
Running
Batch size description
Browse files- translate.py +2 -1
translate.py
CHANGED
@@ -190,7 +190,8 @@ if __name__ == "__main__":
|
|
190 |
"--starting_batch_size",
|
191 |
type=int,
|
192 |
default=128,
|
193 |
-
help="Starting batch size, we will automatically reduce it if we find an OOM error."
|
|
|
194 |
)
|
195 |
|
196 |
parser.add_argument(
|
|
|
190 |
"--starting_batch_size",
|
191 |
type=int,
|
192 |
default=128,
|
193 |
+
help="Starting batch size, we will automatically reduce it if we find an OOM error."
|
194 |
+
"If you use multiple devices, we will divide this number by the number of devices.",
|
195 |
)
|
196 |
|
197 |
parser.add_argument(
|