Awesome
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. 版本历史
更新日期 | 版本 | 版本说明 |
---|---|---|
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时候仅提醒一次) |