• Dmitry Belenko's avatar
    Bugfix for MNASNet (#1224) · 367e8514
    Dmitry Belenko authored
    * Add initial mnasnet impl
    
    * Remove all type hints, comply with PyTorch overall style
    
    * Expose models
    
    * Remove avgpool from features() and add separately
    
    * Fix python3-only stuff, replace subclasses with functions
    
    * fix __all__
    
    * Fix typo
    
    * Remove conditional dropout
    
    * Make dropout functional
    
    * Addressing @fmassa's feedback, round 1
    
    * Replaced adaptive avgpool with mean on H and W to prevent collapsing the batch dimension
    
    * Partially address feedback
    
    * YAPF
    
    * Removed redundant class vars
    
    * Update urls to releases
    
    * Add information to models.rst
    
    * Replace init with kaiming_normal_ in fan-out mode
    
    * Use load_state_dict_from_url
    
    * Fix depth scaling on first 2 layers
    
    * Restore initialization
    
    * Match reference implementation initialization for dense layer
    
    * Meant to use Kaiming
    
    * Remove spurious relu
    
    * Point to the newest 0.5 checkpoint
    
    * Latest pretrained checkpoint
    
    * Restore 1.0 checkpoint
    
    * YAPF
    
    * Implement backwards compat as suggested by Soumith
    
    * Update checkpoint URL
    
    * Move warnings up
    
    * Record a couple more function parameters
    
    * Update comment
    
    * Set the correct version such that if the BC-patched model is saved, it could be reloaded with BC patching again
    
    * Set a member var, not class var
    
    * Update mnasnet.py
    
    Remove unused member var as per review.
    
    * Update the path to weights
    367e8514
mnasnet.py 10.4 KB