Upload train_zimage_lora.py

train_zimage_lora.py

@@ -489,7 +489,7 @@ def main():
                     avg_loss = loss_accumulator / args.gradient_accumulation_steps
                     
                     # 打印格式化的结果
-                    print(f"Steps: {global_step}/{args.max_train_steps}, Avg Loss: {avg_loss:.4f}, LR: {lr_scheduler.get_last_lr()[0]:.6f}")
+                    print(f"Steps: {global_step}/{args.max_train_steps} | Loss: {avg_loss:.4f}")
                     
                     # 清零累加器，为下一轮做准备
                     loss_accumulator = 0.0