轻量级PyTorch通用训练模板pytorch-accelerated解析:2 -- API之Callbacks
简介
这一章将研究pytorch-accelerated的回调机制。
Callback概览
在Trainer中除了可重写的钩子(即自定义训练器的行为)之外,Trainer还包括一个回调系统。
建议将回调Callbacks用于对训练循环的操作不是很重要的代码中,比如日志,但这个决定由用户根据具体的使用情况进行判断。
注意:回调是按顺序执行的,所以如果一个回调被用来修改状态,比如更新一个指标,用户有责任确保这个回调被放在任何将读取这个状态的回调之前(即为了记录的目的)。
回调是在其相应的钩子之后被调用,例如,on_train_epoch_end回调方法在pytorch_accelerated.t