ummagumm-a commited on
Commit
94cf0ee
1 Parent(s): bbe99f6

Upload model

Browse files
Files changed (2) hide show
  1. config.json +2 -2
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -2,8 +2,8 @@
2
  "architectures": [
3
  "AirModelHF"
4
  ],
5
- "device": "cpu",
6
- "dropout_p": 0.01,
7
  "hidden_dim": 50,
8
  "input_dim": 371,
9
  "model_type": "gru",
 
2
  "architectures": [
3
  "AirModelHF"
4
  ],
5
+ "device": "cuda",
6
+ "dropout_p": 0.3,
7
  "hidden_dim": 50,
8
  "input_dim": 371,
9
  "model_type": "gru",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7fc04dd8b403d0f235539f546a4c12e1e4673849aeebc176945e064e2f6de0b0
3
- size 513039
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebb4653ea5bfccbf59233870c34fef8a1c9006e082b7c7a1e2aa0b2937125dca
3
+ size 513167