/efficientnet_pytorch

EfficientNet implemented in Pytorch with SyncBatchNorm layers on multiple GPUs and memory optimization trick for my Kaggle competitions. Code modified from https://github.com/lukemelas/EfficientNet-PyTorch/.

Primary LanguageJupyter Notebook

efficientnet_pytorch