search

ai-computing / aicomp

Other
6 stars 0 forks source link

Feature to send the label to the last stage #7

Open ememos opened 3 months ago

ememos commented 3 months ago

In pipeline parallelism, when the first stage reads data and labels, there is a need to send the labels to the last stage.

ememos commented 3 months ago

move_labels2last_stage function added.