• Min Xu's avatar
    [fix] [FSDP] Do not lose original reshard_after_forward (#880) · 7c2c3e00
    Min Xu authored
    * [fix] [FSDP] Do not lose original reshard_after_forward
    
    - In a corner case we can lose this value
    - Saving it and use it in the reset function fixed it
    - A trivial case probably not worth a dedicated test for now
    
    * added changelog
    7c2c3e00
To find the state of this project's repository at the time of any of these versions, check out the tags..
CHANGELOG.md 16.8 KB