Clean PyTorch implementation of BatchEnsemble Conv1D, Conv2D and Linear layers
Primary LanguagePython