• Sourab Mangrulkar's avatar
    Smangrul/accelerate mp integrate (#23148) · 9f0646a5
    Sourab Mangrulkar authored
    * mixed precision support via accelerate
    
    * fix issues
    
    * fix for the sharded ddp case
    
    * fix flax and tf failing tests
    
    * `refactor the place to create `Accelerator` object
    
    * address comments by removing debugging print statements
    9f0646a5
trainer.py 190 KB