1. 31 Jul, 2019 1 commit
    • Francisco Massa's avatar
      Video reference scripts (#1180) · 5c0b7f31
      Francisco Massa authored
      * Copy classification scripts for video classification
      
      * Initial version of video classification
      
      * add version
      
      * Training of r2plus1d_18 on kinetics work
      
      Gives even slightly better results than expected, with 57.336 top1 clip accuracy. But we count some clips twice in this evaluation
      
      * Cleanups on training script
      
      * Lint
      
      * Minor improvements
      
      * Remove some hacks
      
      * Lint
      5c0b7f31
  2. 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
  3. 08 May, 2019 1 commit
  4. 02 Apr, 2019 2 commits
  5. 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