"Learning Loss for Test-Time Augmentation (NeurIPS 2020)" - View it on GitHub