This commit is contained in:
jingyong hou 2021-12-03 20:43:22 +08:00
parent 48856977b7
commit 325ada3037

View File

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