Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add EfficientNet example using automatic augmentations with DALI (#4678)
This example ports the [EfficientNet](https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Classification/ConvNets/efficientnet/README.md) sample from DeepLearningExamples repository. The example is limited to efficientnet-b0 variant for simplicity. DALI pipeline is updated to use `fn` API and to use new automatic augmentations: adding options to select both AutoAugment and TrivialAugment. The main.py is adjusted so the defaults are suitable for EfficientNet training immediately (previously they were the defaults for RN50 training) and launch.py is no longer needed - the original example was started via launch.py, that looked up default values for specific network in an .yml config and passed them to the main.py. This way we can use main.py directly without the layers of intermediate scripts. The benchmarks from readme are used to implement the L3 test. Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
- Loading branch information