Skip to content

Releases: Tongjilibo/torch4keras

稳定版本v0.1.1

03 Aug 15:32
Compare
Choose a tag to compare

增加指标平滑的SmoothMetricCallback,统一管理指标平滑的问题, 增加SKIP_METRICS,NO_SMOOTH_METRICS,ROUND_PRECISION,默认对指标会进行平滑,修改tensorboard和wandb的callback, 允许跳过nan的指标, Tensorboard可以记录gpu等系统信息

稳定版本v0.1.0.post2

25 Jul 16:09
Compare
Choose a tag to compare

修复v0.1.0的bug,主要是进度条和log的标签平滑的问题

稳定版本v0.1.0

24 Jul 13:36
Compare
Choose a tag to compare

允许调整进度条的显示参数, 进度条和日志同步(如果进度条平滑了则日志也平滑), 自动把tensor转到model.device上, 允许打印第一个batch来检查样本

稳定版本v0.0.9

16 Jul 10:03
Compare
Choose a tag to compare

增加auto_set_cuda_devices自动选择显卡,增加log_info,log_warn, log_error等小函数

稳定版本v0.0.8

25 Jun 15:06
Compare
Choose a tag to compare

增加EmailCallback和WandbCallback, 增加AccelerateTrainer和DeepSpeedTrainer, grad_accumulation_steps内算一个batch,修改Trainer中部分成员函数

稳定版v0.0.7.post3

18 May 14:55
Compare
Choose a tag to compare

修复保存scheduler

稳定版本v0.0.7.post2

17 May 03:15
Compare
Choose a tag to compare

Checkpoint Calback增加保存scheduler, save_weights可自行创建目录,Logger, Tensorboard模块加入lr, 修改predict和add_trainer

稳定版本v0.0.7

05 May 14:06
Compare
Choose a tag to compare

独立出callbacks.py文件, fit允许输入形式为字典,load_weights支持list输入,save_weights支持仅保存可训练参数

稳定版本v0.0.6

12 Feb 15:33
Compare
Choose a tag to compare

增加hf的accelerator测试用例, ddp需要外部控制执行callback, 混合精度支持bf16, 增加add_trainer方法

稳定版本v0.0.5

17 Dec 17:12
Compare
Choose a tag to compare

增加Summary的Callback, 增加Tqdm的进度条展示,保留原有BaseModel的同时,增加Trainer(不从nn.Module继承), 从bert4torch的snippets迁移部分通用函数