|
@@ -309,7 +309,7 @@ def train(model, train_dataloader,eval_dataloader, tokenizer, optimizer, lr_sche
|
|
|
if train_config.save_metrics:
|
|
|
results["metrics_filename"] = metrics_filename
|
|
|
if train_config.flop_counter:
|
|
|
- results["model_flops"]= TFlops
|
|
|
+ results["model_tflops"]= TFlops
|
|
|
#saving the training params including fsdp setting for reference.
|
|
|
if train_config.enable_fsdp and not train_config.use_peft and rank==0:
|
|
|
save_train_params(train_config, fsdp_config, rank)
|