Skip to content

Disable PlottingModules for multi-GPU training#6

Merged
jkiesele merged 2 commits into
jkiesele:mainfrom
kschaeub:main
Apr 28, 2026
Merged

Disable PlottingModules for multi-GPU training#6
jkiesele merged 2 commits into
jkiesele:mainfrom
kschaeub:main

Conversation

@kschaeub

Copy link
Copy Markdown
Contributor

Sets classes inheriting from PlottingModule during model training to None to avoid deepcopy issues in multi-GPU training.

@jkiesele jkiesele merged commit 27a922b into jkiesele:main Apr 28, 2026
4 checks passed
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