• Min Xu's avatar
    [fix]: handle inputs with containers in mixed precision (#486) · 2e9a14e7
    Min Xu authored
    * [fix]: handle inputs with containers
    
    - this is an issue surfaces by vissl as well
    - fix seems to be super simple
    - also cleaned up two tests with respect to multiple such tests
      running back to back (they don't do that presently)
    
    * cleanup
    
    * fix
    
    * lint
    2e9a14e7
test_fsdp_input.py 2.61 KB