• 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
utils.py 3.26 KB