Fix multi-gpu case for `train_cm_ct_unconditional.py` in terms of unwrapping `unet` model #8653
Fix multi-gpu case
df1ad869
tolgacangoz
marked this pull request as draft 1 year ago
Merge branch 'main' into fix-multi-gpu-ct
6aab9bec
Merge branch 'main' into fix-multi-gpu-ct
6c0cca61
tolgacangoz
marked this pull request as ready for review 1 year ago
Merge branch 'main' into fix-multi-gpu-ct
fd5a80ba
Merge branch 'main' into fix-multi-gpu-ct
604a7082
sayakpaul
approved these changes
on 2024-07-14
Prefer previously created `unwrap_model()` function
4b06b954
Merge branch 'main' into fix-multi-gpu-ct
4d8fcec2
a-r-r-o-w
approved these changes
on 2024-07-17
`chore: update unwrap_model() function to use accelerator.unwrap_modeā¦
62f1ff3a
a-r-r-o-w
merged
c1dc2ae6
into main 1 year ago
tolgacangoz
deleted the fix-multi-gpu-ct branch 1 year ago
tolgacangoz
changed the title Fix multi-gpu case for `train_cm_ct_unconditional.py` Fix multi-gpu case for `train_cm_ct_unconditional.py` in terms of unwrapping `unet` model 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub