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

更多版本

5. 更新历史：

更多历史