Skip to content

Model parallelism: Moving labels to the same device as logits for BridgeTower models#22676

Merged
sgugger merged 1 commit intohuggingface:mainfrom shahad-mahmud:model_parallelism_logit_deviceApr 10, 2023
+4

Commits