format
This commit is contained in:
parent
48856977b7
commit
325ada3037
@ -31,4 +31,3 @@ class ElementClassifier(nn.Module):
|
||||
|
||||
def forward(self, x: torch.Tensor):
|
||||
return self.classifier(x)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user