Upload modeling.py with huggingface_hub
Browse files- 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 =
|
| 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
|