❓问题
这是我的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仓库中查看了之前的相关问题,似乎可以进行集成,所以我在想如何解决这个问题。
1条答案
按热度按时间xlpyo6sf1#
嘿,@gursi26,Aim提供了一个专门为PyTorch Lightning设计的内置回调管理器。强烈建议有效地使用此管理器处理所有PyTorch Lightning事件。
在进行这些更改后,您的代码片段将如下所示:
您可以在此页面查看更多详细信息:https://aimstack.readthedocs.io/en/latest/quick_start/integrations.html#integration-with-pytorch-lightning
如有任何其他问题,请随时分享 ☺️