• Vasilis Vryniotis's avatar
    Further enhance Classification Reference (#4444) · c7120163
    Vasilis Vryniotis authored
    * Adding ExponentialLR and LinearLR
    
    * Fix arg type of --lr-warmup-decay
    
    * Adding support of Zero gamma BN and SGD with nesterov.
    
    * Fix --lr-warmup-decay for video_classification.
    
    * Update bn_reinit
    
    * Fix pre-existing bug on num_classes of model
    
    * Remove zero gamma.
    
    * Use fstrings.
    c7120163
train.py 14.5 KB