Releases: Tongjilibo/torch4keras
Releases · Tongjilibo/torch4keras
稳定版本v0.2.9
更新torch.amp
稳定版本v0.2.8
屏蔽torch.load警告, Timeit提示
稳定版本v0.2.7
修改ddp的init_process_group;增加一些available函数,增加get_max_memory;DottableDict支持.赋值和嵌套;增加cachedict
稳定版本v0.2.6
- 增加check_url_available_cached
- 修复Timeit
稳定版本v0.2.5
小修改
- print_table允许中文
- 未安装torch时候仅提醒一次
稳定版本v0.2.4
trainer中可调用nn.Module方法,增加AutoTrainer
稳定版本v0.2.3
- 去除对torch依赖,snippets部分可用;
- 移动bert4torch中snippets
稳定版本v0.2.2
bug修复
- 优化deepspeed的使用
- 修复_prepare_inputs的bug
- 修复断点续训SmoothMetricsCallback起点错误的bug
新增功能
- Trainer断点续训记录batch数
- 增加YamlConfig和IniConfig
稳定版本v0.2.1.post2
修复bug
- 避免训练空数据集error
- 修改run_callback=False的bug
- 修复ddp中mix_precision和torch的ddp重名导致的bug
功能优化
- 默认指标使用滑窗平滑,记录过去interval窗口内的平均指标
- BaseModelDDP修改的更易用,增加init_process_group
- mapping允许是函数(DDP训练的模型等更方便加载)
新增功能
- 训练异常时保存权重;训练异常时候可以保存当前batch数据用于debug
- Checkpoint的Callback增加save_on_train_end
- 增加SystemCallback用于监控gpu, cpu,内存等占用
稳定版本v0.2.0
fit中修改.train()逻辑较少耗时