• Francisco Massa's avatar
    Initial version of classification reference scripts (#819) · 27ff89f6
    Francisco Massa authored
    * Initial version of classification reference training script
    
    * Updates
    
    * Minor updates
    
    * Expose a few more options
    
    * Load optimizer and lr_scheduler when resuming
    
    Also log the learning rate
    
    * Evaluation-only and minor improvements
    
    Identified a bug in the reporting of the results. They need to be reduced between all processes
    
    * Address Soumith's comment
    
    * Fix some approximations on the evaluation metric
    
    * Flake8
    27ff89f6
utils.py 5.74 KB