- 移除 state_dict_fn 参数 - _save_checkpoint 中先 unwrap_model 再 state_dict() |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| metric_util.py | ||
| optim.py | ||
| schedule.py | ||
| strategy.py | ||
| train_callback.py | ||
| train_context.py | ||
| trainer.py | ||
- 移除 state_dict_fn 参数 - _save_checkpoint 中先 unwrap_model 再 state_dict() |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| metric_util.py | ||
| optim.py | ||
| schedule.py | ||
| strategy.py | ||
| train_callback.py | ||
| train_context.py | ||
| trainer.py | ||