Torch4keras

Latest version: v0.2.2

Safety actively analyzes 626004 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

Page 1 of 5

0.2.2

bug修复
- 优化deepspeed的使用
- 修复_prepare_inputs的bug
- 修复断点续训SmoothMetricsCallback起点错误的bug

新增功能
- Trainer断点续训记录batch数
- 增加YamlConfig和IniConfig

0.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,内存等占用

0.2.0

fit中修改.train()逻辑较少耗时

0.1.9

增加Timeit, Timeit2, timeid等时间/速度监控

0.1.8

重新整理snippets, 重写save_pretrained

0.1.7

- 增加SimpleStreamFileLogger和LoggerHandler
- 修改Logger的格式

Page 1 of 5

© 2024 Safety CLI Cybersecurity Inc. All Rights Reserved.