AstrAI

Commit Graph

Author	SHA1	Message	Date
ViperEkura	08dde46778	fix: 修复训练循环 step/backward 顺序，重构为三重循环嵌套 - 训练循环改用 itertools.batched 实现 epoch→step→batch 三重嵌套 - on_step_begin 包裹 batch 循环，on_step_end 后接 optimizer.step/scheduler.step - 修复首次 iteration=0 时 optimizer.step() 在 backward 之前触发的 bug - GradientClippingCallback 改为 on_step_end（梯度已累积，step 前裁剪） - SchedulerCallback 移除，schduler.step 由 trainer 在 optimizer.step 后直接调用 - metric_util 提取 _grad_stat 公共 helper，if param.grad: 修正为 is not None	2026-05-15 14:44:44 +08:00
ViperEkura	0852b852f8	refactor: 优化参数传递，清理导入样式	2026-04-03 22:06:32 +08:00
ViperEkura	2e009cf59a	chore: 更新项目名称	2026-03-31 09:34:11 +08:00

Author

SHA1

Message

Date

ViperEkura

08dde46778

fix: 修复训练循环 step/backward 顺序，重构为三重循环嵌套

- 训练循环改用 itertools.batched 实现 epoch→step→batch 三重嵌套
- on_step_begin 包裹 batch 循环，on_step_end 后接 optimizer.step/scheduler.step
- 修复首次 iteration=0 时 optimizer.step() 在 backward 之前触发的 bug
- GradientClippingCallback 改为 on_step_end（梯度已累积，step 前裁剪）
- SchedulerCallback 移除，schduler.step 由 trainer 在 optimizer.step 后直接调用
- metric_util 提取 _grad_stat 公共 helper，if param.grad: 修正为 is not None

2026-05-15 14:44:44 +08:00

ViperEkura

0852b852f8

refactor: 优化参数传递，清理导入样式

2026-04-03 22:06:32 +08:00

ViperEkura

2e009cf59a

chore: 更新项目名称

2026-03-31 09:34:11 +08:00

3 Commits