-
Notifications
You must be signed in to change notification settings - Fork 5.8k
Open
Labels
PFCCPaddle Framework Contributor Club,https://github.com/PaddlePaddle/community/tree/master/pfccPaddle Framework Contributor Club,https://github.com/PaddlePaddle/community/tree/master/pfccgood first issuestatus/following-up跟进中跟进中type/feature-request新需求申请新需求申请
Description
需求描述 Feature Description
当前paddle.Callback仅支持读取metric,lr,loss数值等日志性记录,希望支持读取每个iter的输入数据、模型输出
参考PyTorch lighting的实现
on_predict_batch_end(trainer, pl_module, outputs, batch, batch_idx, dataloader_idx)
替代实现 Alternatives
修改paddle.Model实现,增加一个缓冲区,记录forward函数的输入及输出。然后在callback中由self.model.input和self.model.output读取
Metadata
Metadata
Assignees
Labels
PFCCPaddle Framework Contributor Club,https://github.com/PaddlePaddle/community/tree/master/pfccPaddle Framework Contributor Club,https://github.com/PaddlePaddle/community/tree/master/pfccgood first issuestatus/following-up跟进中跟进中type/feature-request新需求申请新需求申请