1. 10 May, 2019 1 commit
    • Francisco Massa's avatar
      Initial version of segmentation reference scripts (#820) · 50d54a82
      Francisco Massa authored
      * Initial version of the segmentation examples
      
      WIP
      
      * Cleanups
      
      * [WIP]
      
      * Tag where runs are being executed
      
      * Minor additions
      
      * Update model with new resnet API
      
      * [WIP] Using torchvision datasets
      
      * Improving datasets
      
      Leverage more and more torchvision datasets
      
      * Reorganizing datasets
      
      * PEP8
      
      * No more SegmentationModel
      
      Also remove outplanes from ResNet, and add a function for querying intermediate outputs. I won't keep it in the end, because it's very hacky and don't work with tracing
      
      * Minor cleanups
      
      * Moving transforms to its own file
      
      * Move models to torchvision
      
      * Bugfixes
      
      * Multiply LR by 10 for classifier
      
      * Remove classifier x 10
      
      * Add tests for segmentation models
      
      * Update with latest utils from classification
      
      * Lint and missing import
      50d54a82
  2. 08 May, 2019 1 commit
  3. 02 Apr, 2019 2 commits
  4. 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