• Pete's avatar
    Use original forward pass directly when in eval mode from within checkpoint wrapper (#709) · 370b8483
    Pete authored
    * add failing test
    
    * add fix
    
    * use 'torch.is_grad_enabled()' instead of 'module.training'
    
    * Revert "add failing test"
    
    This reverts commit 1c34242208f9b2c5fa6c8f181434c2be6d7cdbc0.
    
    * add simple test
    
    * improve test
    
    * add check for fwd_counter
    
    * revert typing/format changes
    
    * move to new test file
    
    * CHANGELOG
    
    * remove old test
    
    * fix import order
    
    * fix test to be compat with torch 1.6.0
    
    * clean up
    
    * comments
    
    * isort 🤦
    370b8483
To find the state of this project's repository at the time of any of these versions, check out the tags..
CHANGELOG.md 9.77 KB