1. 10 Feb, 2021 1 commit
  2. 31 Mar, 2020 1 commit
  3. 14 Jun, 2019 1 commit
    • LXYTSOS's avatar
      utils.py in references can't work with pytorch-cpu (#1023) · 250bac89
      LXYTSOS authored
      * can't work with pytorch-cpu fixed
      
      utils.py can't work with pytorch-cpu because of this line of code `memory=torch.cuda.max_memory_allocated()`
      
      * can't work with pytorch-cpu fixed
      
      utils.py can't work with pytorch-cpu because of this line of code 'memory=torch.cuda.max_memory_allocated()'
      250bac89
  4. 08 May, 2019 1 commit
  5. 02 Apr, 2019 2 commits
  6. 28 Mar, 2019 1 commit
    • 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