• Vasilis Vryniotis's avatar
    Add SSDlite architecture with MobileNetV3 backbones (#3757) · 43d77206
    Vasilis Vryniotis authored
    * Partial implementation of SSDlite.
    
    * Add normal init and BN hyperparams.
    
    * Refactor to keep JIT happy
    
    * Completed SSDlite.
    
    * Fix lint
    
    * Update todos
    
    * Add expected file in repo.
    
    * Use C4 expansion instead of C4 output.
    
    * Change scales formula for Default Boxes.
    
    * Add cosine annealing on trainer.
    
    * Make T_max count epochs.
    
    * Fix test and handle corner-case.
    
    * Add support of support width_mult
    
    * Add ssdlite presets.
    
    * Change ReLU6, [-1,1] rescaling, backbone init & no pretraining.
    
    * Use _reduced_tail=True.
    
    * Add sync BN support.
    
    * Adding the best config along with its weights and documentation.
    
    * Make mean/std configurable.
    
    * Fix not implemented for half exception
    43d77206
train.py 9.73 KB