Batchnorm2d

[Millebean]

Hi, I am testing out the model with amazon_webcam dataset. The model returns an error saying 'expected 4D input (got 2D input)' from models.py in the line domain_out = self.domain_classifier(reverse_feature).

I figured it should be batchnorm1D instead of batchnorm2D in the domain_classifier.