0
代码介绍
代码
Issues
Pull Requests
流水线
Actions
讨论
Wiki
项目成员
分析
项目设置
0
  1. mindformers
  2. /
  3. tests
  4. /
  5. st
  6. /
  7. test_ut
  8. /
  9. test_trainer
huangjingweihuangjingwei[master][refactor] Adapt ms context setting & getting.
06a6ac04创建于 3月5日历史提交
文件最后提交记录最后更新时间
__init__.py
feat: add PyNative mode trainer with callback system and comprehensive tests - Add Trainer class supporting PyNative mode with train(), eval(), and get_batch() methods - Implement callback system with CallbackHandler and TrainerCallback base class - Add LossCallback for loss monitoring and logging - Add CheckpointCallback for model checkpoint management - Add TrainerState for tracking training state - Implement comprehensive unit tests for all components - Support pretrain and finetune training modes - Add distributed training support with data parallelism - Fix all pylint code style issues 4 个月前
test_train_state.py
[Refactor]Refactor pynative structure: Move trainer and callback modules to mindformers/pynative and update test imports. 4 个月前
test_trainer.py
[Refactor]Refactor pynative structure: Move trainer and callback modules to mindformers/pynative and update test imports. 4 个月前
test_trainer_config.yaml
add testcase for trainer 6 个月前
test_trainer_methods.py
[master][refactor] Adapt ms context setting & getting. 2 个月前
test_trainstate_integration.py
[Refactor]Refactor pynative structure: Move trainer and callback modules to mindformers/pynative and update test imports. 4 个月前