han-xudong commited on
Commit
aaaa795
·
verified ·
1 Parent(s): 9bd5849

Upload modeling.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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.Module()
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(