AimLogger在PyTorch lightning中无法使用,

yk9xbfzb 于 8个月前发布在其他

关注(0)|答案(1)|浏览(377)

❓问题

这是我的PyTorch lightning训练器的代码：

import lightning as L
trainer = L.Trainer(
    max_epochs=EPOCHS,
    accelerator=DEVICE,
    logger = aim.Run(experiment="trial_exp_lightning")
)
trainer.fit(training_module, train_loader, test_loader)

运行这段代码后，我得到了以下错误信息。

AttributeError: 'Run' object has no attribute 'save_dir'

我已经在lightning和aim仓库中查看了之前的相关问题，似乎可以进行集成，所以我在想如何解决这个问题。

aim

来源：https://github.com/aimhubio/aim/issues/3088

1条答案

按热度按时间

xlpyo6sf1#

嘿，@gursi26,Aim提供了一个专门为PyTorch Lightning设计的内置回调管理器。强烈建议有效地使用此管理器处理所有PyTorch Lightning事件。
在进行这些更改后，您的代码片段将如下所示：

import lightning as L
aim_logger = AimLogger(
    experiment_name="trial_exp_lightning",
)
trainer = L.Trainer(
    max_epochs=EPOCHS,
    accelerator=DEVICE,
    logger = aim_logger
)
trainer.fit(training_module, train_loader, test_loader)

您可以在此页面查看更多详细信息：https://aimstack.readthedocs.io/en/latest/quick_start/integrations.html#integration-with-pytorch-lightning
如有任何其他问题，请随时分享 ☺️

展开查看全部

赞(0）回复(0）举报 8个月前

我来回答

AimLogger在PyTorch lightning中无法使用,

❓问题

1条答案

相关问题

热门标签

最新问答