🚀 功能

在 Aim's pytorch_lightning 适配器中添加 epoch,只要它可用。

介绍

通常，当前的 epoch 编号可以从由 pytorch_lightning 发送的日志中获取为 metric。如果我们可以将这个 epoch 提取出来传递给 Aim 的记录器，那么人们就可以在 Aim 的度量资源管理器中将 x 轴设置为 epoch。这有助于在具有可能不同批处理大小的不同实验之间比较记录的值。

其他上下文

我有兴趣实现这个功能(应该像在适配器中添加 epoch=metric.get('epoch', None) 一样简单),但我不确定这是否对你们感兴趣？此外，我认为最好还是将 'epoch' 保留在指标内部(而不是在获取到它之后将其放在 del metric['epoch'] 中)。