Releases: Tongjilibo/torch4keras
Releases · Tongjilibo/torch4keras
稳定版本v0.0.4
为callback增加on_train_step_end方法, 修复BaseModel(net)方式的bug
修复版本v0.0.3.post
修复DDP下只有master_rank打印
稳定版本v0.0.3
参考Keras修改了callback的逻辑
稳定版本v0.0.2
增加Checkpoint, Evaluator等自带Callback, 修改BaseModel(net)方式
v0.0.1
初始版本,从bert4torch中剥离出来的普通Trainer,支持单机多卡,混合精度,梯度裁剪,梯度累积等