feat: add cur_time to log
This commit is contained in:
parent
8d6f544698
commit
cba294f74d
|
@ -1,5 +1,7 @@
|
|||
#!/bin/bash
|
||||
|
||||
|
||||
# bash run_once.sh lora_sft Qwen-7B 8 50
|
||||
run_type="$1"
|
||||
model="$2"
|
||||
gpu_cnt="$3"
|
||||
|
|
|
@ -303,6 +303,7 @@ class LogCallback(TrainerCallback):
|
|||
learning_rate=state.log_history[-1].get("learning_rate", None),
|
||||
epoch=state.log_history[-1].get("epoch", None),
|
||||
percentage=round(self.cur_steps / self.max_steps * 100, 2) if self.max_steps != 0 else 100,
|
||||
cur_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),
|
||||
elapsed_time=self.elapsed_time,
|
||||
remaining_time=self.remaining_time,
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue