Hajime MATSUMOTO commited on
Commit
29f5c51
·
1 Parent(s): 702c22f

Fix f-string format specifier error in logging callback

Browse files
Files changed (1) hide show
  1. train.py +5 -2
train.py CHANGED
@@ -104,10 +104,13 @@ class VerboseLoggingCallback(TrainerCallback):
104
  loss = logs.get("loss", "N/A")
105
  lr = logs.get("learning_rate", "N/A")
106
 
 
 
 
107
  print(f"[{datetime.now().strftime('%H:%M:%S')}] "
108
  f"Step {state.global_step}/{state.max_steps} ({progress:.1f}%) | "
109
- f"Loss: {loss:.4f if isinstance(loss, float) else loss} | "
110
- f"LR: {lr:.2e if isinstance(lr, float) else lr} | "
111
  f"Elapsed: {elapsed_str} | ETA: {eta_str}")
112
 
113
  # GPU メモリ使用量(10ステップごと)
 
104
  loss = logs.get("loss", "N/A")
105
  lr = logs.get("learning_rate", "N/A")
106
 
107
+ loss_str = f"{loss:.4f}" if isinstance(loss, float) else str(loss)
108
+ lr_str = f"{lr:.2e}" if isinstance(lr, float) else str(lr)
109
+
110
  print(f"[{datetime.now().strftime('%H:%M:%S')}] "
111
  f"Step {state.global_step}/{state.max_steps} ({progress:.1f}%) | "
112
+ f"Loss: {loss_str} | "
113
+ f"LR: {lr_str} | "
114
  f"Elapsed: {elapsed_str} | ETA: {eta_str}")
115
 
116
  # GPU メモリ使用量(10ステップごと)