Skip to content

允许Callbacks接收输入数据及模型输出 #46996

@parap1uie-s

Description

@parap1uie-s

需求描述 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

Labels

PFCCPaddle Framework Contributor Club,https://github.com/PaddlePaddle/community/tree/master/pfccgood first issuestatus/following-up跟进中type/feature-request新需求申请

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions