Upload model

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e53058338e26df015dae90d11cbe0ef991222907bd39bb4476f56271b2ce09bb
-size 44340
+oid sha256:ffbd1a93fbaebb531e4f00b86e40eb0d7f16f67dd143a5bf99bf5606f7666dee
+size 44272

modeling_IQtransformer.py

@@ -222,7 +222,7 @@ class transformerModel(PreTrainedModel):
                                               config.num_heads, config.dropout))
 
         self.l1 = nn.Linear(64, 16)
-        self.l2 = nn.Linear(16, 5)
+        self.l2 = nn.Linear(16, 4)
 
     def forward(self, X, valid_lens, *args):
         # 因为位置编码值在-1和1之间，