Update classifier.py

This commit is contained in:
xiaohou 2021-11-18 16:06:59 +08:00 committed by GitHub
parent 2dff1fd5fd
commit a7f54824a3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -30,4 +30,4 @@ class ElementClassifier(nn.Module):
self.classifier = classifier
def forward(self, x: torch.Tensor):
return self.classifier(x)
return self.classifier(x)