han-xudong commited on
Commit
4fa28d2
·
verified ·
1 Parent(s): da98f71

Upload modeling.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling.py +1 -1
modeling.py CHANGED
@@ -48,7 +48,7 @@ class FingerNet(PreTrainedModel):
48
  outputs = []
49
  for i in range(len(self.y_dim)):
50
  # Get the estimator for the i-th output
51
- estimator = getattr(self, f"estimator_{i}")
52
  y = estimator(x)
53
  outputs.append(y)
54
  return outputs
 
48
  outputs = []
49
  for i in range(len(self.y_dim)):
50
  # Get the estimator for the i-th output
51
+ estimator = self.model[f"estimator_{i}"]
52
  y = estimator(x)
53
  outputs.append(y)
54
  return outputs