Skip to content

Fix multi gpu training for deepedit + publish model#494

Merged
SachidanandAlle merged 1 commit intomainfrom
multi_gpu_fix
Nov 5, 2021
Merged

Fix multi gpu training for deepedit + publish model#494
SachidanandAlle merged 1 commit intomainfrom
multi_gpu_fix

Conversation

@SachidanandAlle
Copy link
Copy Markdown
Collaborator

Signed-off-by: Sachidanand Alle [email protected]

  • Fix multi gpu training for DynamicUNet (enable find_unused_parameters flag for DDP)
  • Fix to publish best metric model (mis-match in key metric filename)
  • Rollback default spatial size for Deepedit model to [128,128,128] for faster training

@SachidanandAlle SachidanandAlle self-assigned this Nov 5, 2021
@SachidanandAlle SachidanandAlle added the 0.3.0 Targeted to Release version 0.3 label Nov 5, 2021
@SachidanandAlle SachidanandAlle linked an issue Nov 5, 2021 that may be closed by this pull request
@SachidanandAlle SachidanandAlle merged commit 0dd17a2 into main Nov 5, 2021
@SachidanandAlle SachidanandAlle deleted the multi_gpu_fix branch November 5, 2021 12:44
Douwe-Spaanderman pushed a commit to Douwe-Spaanderman/MONAILabel that referenced this pull request Dec 9, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

0.3.0 Targeted to Release version 0.3

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Deepedit/DynUNetV1 not supporting MULTI GPU training

2 participants