🚀 功能
在 Aim's pytorch_lightning
适配器中添加 epoch
,只要它可用。
介绍
通常,当前的 epoch 编号可以从由 pytorch_lightning
发送的日志中获取为 metric
。如果我们可以将这个 epoch
提取出来传递给 Aim 的记录器,那么人们就可以在 Aim 的度量资源管理器中将 x 轴设置为 epoch。这有助于在具有可能不同批处理大小的不同实验之间比较记录的值。
其他上下文
我有兴趣实现这个功能(应该像在适配器中添加 epoch=metric.get('epoch', None)
一样简单),但我不确定这是否对你们感兴趣?此外,我认为最好还是将 'epoch' 保留在指标内部(而不是在获取到它之后将其放在 del metric['epoch']
中)。
1条答案
按热度按时间arknldoa1#
嘿,@we-taper 感谢你的举报,抱歉回复晚了。🙏
你对贡献这个感兴趣吗?