Skip to content

GitLab

".github/workflows/windows_release_cu118_dependencies.yml" did not exist on "09666f93cfc70526d74149ac2ad9a0c83516d943"
Switch branch/tag
Find file BlameHistoryPermalink
  • Julien Chaumond's avatar
    Fix nn.DataParallel compatibility in PyTorch 1.5 (#4300) · 4c068936
    Julien Chaumond authored 
    * Test case for #3936

* multigpu tests pass on pytorch 1.4.0

* Fixup

* multigpu tests pass on pytorch 1.5.0

* Update src/transformers/modeling_utils.py

* Update src/transformers/modeling_utils.py

* rename multigpu to require_multigpu

* mode doc
    4c068936
test_modeling_gpt2.py 15.6 KB