Update classifier.py
This commit is contained in:
parent
2dff1fd5fd
commit
a7f54824a3
@ -30,4 +30,4 @@ class ElementClassifier(nn.Module):
|
|||||||
self.classifier = classifier
|
self.classifier = classifier
|
||||||
|
|
||||||
def forward(self, x: torch.Tensor):
|
def forward(self, x: torch.Tensor):
|
||||||
return self.classifier(x)
|
return self.classifier(x)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user