use locale agnostic seperator to make large nums easier to read (#1503)
Browse files
src/axolotl/utils/trainer.py
CHANGED
@@ -198,7 +198,7 @@ def calculate_total_num_steps(cfg, train_dataset, update=True):
|
|
198 |
.apply(lambda x: len(x)) # pylint: disable=unnecessary-lambda
|
199 |
.values
|
200 |
)
|
201 |
-
LOG.debug(f"total_num_tokens: {total_num_tokens}", main_process_only=True)
|
202 |
if update:
|
203 |
cfg.total_num_tokens = total_num_tokens
|
204 |
|
@@ -212,7 +212,7 @@ def calculate_total_num_steps(cfg, train_dataset, update=True):
|
|
212 |
.sum()
|
213 |
)
|
214 |
LOG.debug(
|
215 |
-
f"`total_supervised_tokens: {total_supervised_tokens}`",
|
216 |
main_process_only=True,
|
217 |
)
|
218 |
if update:
|
@@ -239,7 +239,7 @@ def calculate_total_num_steps(cfg, train_dataset, update=True):
|
|
239 |
* cfg.num_epochs
|
240 |
)
|
241 |
LOG.debug(
|
242 |
-
f"total_num_tokens: {cfg.total_num_tokens}, total_num_steps: {total_num_steps}",
|
243 |
main_process_only=True,
|
244 |
)
|
245 |
else:
|
|
|
198 |
.apply(lambda x: len(x)) # pylint: disable=unnecessary-lambda
|
199 |
.values
|
200 |
)
|
201 |
+
LOG.debug(f"total_num_tokens: {total_num_tokens:_}", main_process_only=True)
|
202 |
if update:
|
203 |
cfg.total_num_tokens = total_num_tokens
|
204 |
|
|
|
212 |
.sum()
|
213 |
)
|
214 |
LOG.debug(
|
215 |
+
f"`total_supervised_tokens: {total_supervised_tokens:_}`",
|
216 |
main_process_only=True,
|
217 |
)
|
218 |
if update:
|
|
|
239 |
* cfg.num_epochs
|
240 |
)
|
241 |
LOG.debug(
|
242 |
+
f"total_num_tokens: {cfg.total_num_tokens:_}, total_num_steps: {total_num_steps:_}",
|
243 |
main_process_only=True,
|
244 |
)
|
245 |
else:
|