Update classifier.py
This commit is contained in:
parent
2dff1fd5fd
commit
a7f54824a3
@ -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)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user