Upload model

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:927e38d4306b21e0833edf180d3b2d59c94bd0997a88b3c25289e973417c7370
-size 44340
+oid sha256:59ddb6814299930ea3c647584bc2aeb3c9d3bb0cb3d8bdf4036eab6c9fac9adb
+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之间，