winglian commited on
Commit
da9b1a3
·
unverified ·
1 Parent(s): 057fa44

use locale agnostic seperator to make large nums easier to read (#1503)

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/trainer.py +3 -3
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: