Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Set ImageNet data augmentation by default #13757

Closed
wants to merge 3 commits into from
Closed

Commits on Jan 2, 2019

  1. Set ImageNet data augmentation by default

    https://github.com/apache/incubator-mxnet/blob/a38278ddebfcc9459d64237086cd7977ec20c70e/example/image-classification/train_imagenet.py#L42
    
    When I try to train imagenet with this line commented, the train-accuracy reaches 99% while the validation-accuracy is only less than 50% (single machine, 8 GPUs, global batchsize=2048, Resnet50). Absolutely this is overfitting.
    
    Then I uncomment this line and try again with the same experiment settings. This time both train and validation accuracy converge to about 70%. 
    
    Thus, it seems that this data augmentation is pretty important for ImageNet training. Perhaps it will be better to uncomment this as default, so that future developers won't get confused by the over-fit issue.
    ymjiang authored Jan 2, 2019
    Configuration menu
    Copy the full SHA
    42c4dc7 View commit details
    Browse the repository at this point in the history

Commits on Mar 19, 2019

  1. Configuration menu
    Copy the full SHA
    52f2945 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    bd4faf0 View commit details
    Browse the repository at this point in the history