[ICLR 2022] Official pytorch implementation of "Uncertainty Modeling for Out-of-Distribution Generalization" in International Conference on Learning Representations (ICLR) 2022.
142
stars
16
forks
source link
error about "class DistributionUncertainty(nn.Module)" #9
self.pertubration0 = pertubration(dim=64, p=uncertainty) if pertubration else nn.Identity() TypeError: init() got an unexpected keyword argument 'dim'
I would like to ask, why do I get this error message when initializing the DSU module?