• Y
    Revert to mixed precision training with manual control (#3434) · efc994ec
    Yang Zhang 提交于
    * Place mixed precision inside PaddleDetection
    
    roll back to the monkey patch version as a temporary measure, before it is
    merged into paddle
    
    * Add command flag for `loss_scale`
    
    * Fix a stupid indentation error
    
    optimizer should be in the mixed precision context
    
    * Initial FP16 training
    
    * Add mixed precision training to rest of the detection models
    
    * Revert "Add support for mixed precision training (#3406)"
    
    This reverts commit 3a2c1062.
    
    * Bug fixes and some tweaks
    efc994ec
train.py 11.7 KB