SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Perceiver] Skip multi-gpu tests for now (#14813)

Browse Source 
* [Perceiver] Skip multi-gpu tests for now

* Update tests/test_modeling_perceiver.py

* up

* up
This commit is contained in:
Patrick von Platen
2021-12-20 15:22:50 +01:00
committed by GitHub
parent 8a818c26cb
commit 952a77b05d
7 changed files with 28 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/source/model_doc/perceiver.mdx
View File

@@ -86,6 +86,10 @@ is implemented in the library. Note that the models available in the library onl
with the Perceiver. There are many more use cases, including question answering, named-entity recognition, object detection,
audio classification, video classification, etc.
**Note**:
- Perceiver does **not** work with `torch.nn.DataParallel` due to a bug in PyTorch, see [issue #36035](https://github.com/pytorch/pytorch/issues/36035)
## Perceiver specific outputs
[[autodoc]] models.perceiver.modeling_perceiver.PerceiverModelOutput
@@ -208,4 +212,4 @@ audio classification, video classification, etc.
## PerceiverForMultimodalAutoencoding
[[autodoc]] PerceiverForMultimodalAutoencoding
- forward
- forward