torch4keras

bert4torch底层训练框架,用keras风格写torch代码

APACHE-2.0 License

Downloads
479
Stars
74
Committers
2

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

  • 初衷:前期功能是作为bert4torchrec4torch的Trainer

  • 训练:

3. 快速上手

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时候仅提醒一次)

更多版本

5. 更新历史:

更多历史

Package Rankings
Top 14.23% on Pypi.org
Top 6.68% on Proxy.golang.org
Badges
Extracted from project README
licence GitHub release PyPI PyPI - Downloads GitHub stars GitHub Issues contributions welcome
Related Projects