Spaces:
Runtime error
Runtime error
Hajime MATSUMOTO
commited on
Commit
·
29f5c51
1
Parent(s):
702c22f
Fix f-string format specifier error in logging callback
Browse files
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: {
|
| 110 |
-
f"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ステップごと)
|