This commit is contained in:
游雁 2024-05-20 16:20:45 +08:00
parent 1ff66c5260
commit 47c14aff79

View File

@ -712,8 +712,8 @@ class Trainer:
"data_split_i": kwargs.get("data_split_i", 0),
"data_split_num": kwargs.get("data_split_num", 1),
"log_step": batch_idx + kwargs.get("start_step", 0),
"batch_total": batch_idx,
"step_in_epoch": batch_idx,
"batch_total": batch_idx + 1,
"step_in_epoch": batch_idx + 1,
"lr": 0.0,
}