format
This commit is contained in:
parent
48856977b7
commit
325ada3037
@ -31,4 +31,3 @@ class ElementClassifier(nn.Module):
|
|||||||
|
|
||||||
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