src/transformers/models/bridgetower/modeling_bridgetower.py · 870d91fb89dd2e0924df6ad2fe85e3084f1c380f · chenpangpang / transformers · GitLab

"tests/models/xlm/__init__.py" did not exist on "5b396457e5035a8b16ddee14b205c098598fe6bb"

Find file Blame History Permalink

Model parallelism: Moving labels to the same device as logits for BridgeTower models (#22676) · 870d91fb
Shahad Mahmud authored Apr 10, 2023
```
BrideTower Model parallelism logits device for loss calculation
```
870d91fb

modeling_bridgetower.py 86 KB