README.md
May 13, 2026 · View on GitHub
Documentation | Bert4torch | Examples | Source code | build_MiniLLM_from_scratch
1. 下载安装
安装稳定版
pip install torch4keras
安装最新版
pip install git+https://github.com/Tongjilibo/torch4keras.git
2. 功能
-
简述：抽象出来的Trainer，适用于一般神经网络的训练，仅需关注网络结构代码
-
特色：进度条展示训练过程，自定义metric，自带Evaluator, Checkpoint, Tensorboard, Logger等Callback，也可自定义Callback
-
初衷：前期功能是作为bert4torch和rec4torch的Trainer
-
训练：
3. 快速上手
- 参考bert4torch的训练过程
- 简单示例: turorials_mnist
4. 版本历史
| 更新日期 | 版本 | 版本说明 |
|---|---|---|
| 20260513 | v0.3.4 | 把transfromers的logging模块迁移，修改名称为DotDict |
| 20260114 | v0.3.3 | 修改功能函数（DottableDict） |
| 20250925 | v0.3.2 | 增加常见 shell命令 |
| 20250721 | v0.3.1 | add cache_text, try_except, TryExcept, 完善耗时计算，修改 load_checkpoint，增加 green等颜色函数 |
| 20250615 | v0.3.0 | 增加 safe_torch_load和 safe_import |
| 20250511 | v0.2.9.post2 | trainer增加kwargs |
| 20250401 | v0.2.9 | 更新torch.amp |
| 20250211 | v0.2.8 | 屏蔽torch.load警告, Timeit提示 |
| 20240928 | v0.2.7 | 修改ddp的init_process_group;增加一些available函数,增加get_max_memory;DottableDict支持.赋值和嵌套;增加cachedict |
| 20240814 | v0.2.6 | 小修改(增加check_url_available_cached, 修复Timeit) |
| 20240730 | v0.2.5 | 小修改(print_table允许中文, 未安装torch时候仅提醒一次) |