Accueil Explorer Aide
S'inscrire Connexion
marx
/
llama-recipes
1
0
Fork 0
Fichiers Tickets 0 Pull Requests 0 Wiki
Parcourir la source

remove the redundant lr step

Hamid Shojanazeri il y a 1 an
Parent
31d6ce8bf6
commit
4f70348b94
1 fichiers modifiés avec 0 ajouts et 2 suppressions
Vue unifiée Afficher les stats Diff
  1. 0 2
      utils/train_utils.py

+ 0 - 2
utils/train_utils.py
Voir le fichier 

@@ -193,8 +193,6 @@ def train(model, train_dataloader,eval_dataloader, tokenizer, optimizer, lr_sche
 
         else:
 
         else:
 
             print(f"Epoch {epoch+1}: train_perplexity={train_perplexity:.4f}, train_epoch_loss={train_epoch_loss:.4f}")
 
             print(f"Epoch {epoch+1}: train_perplexity={train_perplexity:.4f}, train_epoch_loss={train_epoch_loss:.4f}")
 
             
             
-        lr_scheduler.step()
 
-
 
     avg_train_prep = sum(train_prep)/len(train_prep)
 
     avg_train_prep = sum(train_prep)/len(train_prep)
 
     avg_train_loss = sum(train_loss)/len(train_loss)
 
     avg_train_loss = sum(train_loss)/len(train_loss)
 
     if train_config.run_validation:
 
     if train_config.run_validation: