Upload modeling.py with huggingface_hub
Browse files- modeling.py +1 -1
modeling.py
CHANGED
|
@@ -30,7 +30,7 @@ class FingerNet(PreTrainedModel):
|
|
| 30 |
self.h1_dim = config.h1_dim
|
| 31 |
self.h2_dim = config.h2_dim
|
| 32 |
|
| 33 |
-
self.model = nn.
|
| 34 |
# Define the model architecture
|
| 35 |
for i in range(len(self.y_dim)):
|
| 36 |
setattr(
|
|
|
|
| 30 |
self.h1_dim = config.h1_dim
|
| 31 |
self.h2_dim = config.h2_dim
|
| 32 |
|
| 33 |
+
self.model = nn.ModuleDict()
|
| 34 |
# Define the model architecture
|
| 35 |
for i in range(len(self.y_dim)):
|
| 36 |
setattr(
|