Skip to content

Conversation

ShaohonChen
Copy link
Contributor

修复使用pytorch lightning的swanlab日志集成时,使用多卡训练会出现self. _experiment未声明问题。

修复方法

在pytorch lightning的日志收集器逻辑中,初始化日志记录器只在master进程中调用,因此要在init中预先将self. _experiment赋值为none

@ShaohonChen ShaohonChen requested a review from Zeyi-Lin February 3, 2025 03:26
@ShaohonChen ShaohonChen self-assigned this Feb 3, 2025
@Zeyi-Lin Zeyi-Lin merged commit 50c92be into main Feb 3, 2025
5 checks passed
@Zeyi-Lin Zeyi-Lin deleted the fix-pytl-ddp-bug branch February 5, 2025 05:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants