Kaynağa Gözat

fixed a typo

Kai Wu 1 yıl önce
ebeveyn
işleme
7b1a9413d2
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. 1 1
      src/llama_recipes/utils/train_utils.py

+ 1 - 1
src/llama_recipes/utils/train_utils.py

@@ -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)