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:
|