transformers
151425dd - Model parallelism: Moving labels to same devices as the logits are (#22691)

Commit

2 years ago

Model parallelism: Moving labels to same devices as the logits are (#22691) Model parallelism correct labels device

References

Author

shahad-mahmud

Parents