WarriorsSami commited on
Commit
fe3f1e2
·
1 Parent(s): 2d4eafe

feat: add first model version

Browse files
.gitignore CHANGED
@@ -1,3 +1,4 @@
1
  target
2
- sam-artifacts/
3
- !sam-artifacts/model
 
 
1
  target
2
+ sam-artifacts/**
3
+ !sam-artifacts/model.mpk
4
+ !sam-artifacts/config.json
sam-artifacts/config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "transformer": {
3
+ "d_model": 256,
4
+ "d_ff": 1024,
5
+ "n_heads": 8,
6
+ "n_layers": 4,
7
+ "dropout": 0.1,
8
+ "norm_first": true,
9
+ "quiet_softmax": true,
10
+ "initializer": {
11
+ "KaimingUniform": {
12
+ "gain": 0.5773502691896258,
13
+ "fan_out_only": false
14
+ }
15
+ }
16
+ },
17
+ "optimizer": {
18
+ "weight_decay": {
19
+ "penalty": 0.00005
20
+ },
21
+ "grad_clipping": null,
22
+ "beta_1": 0.9,
23
+ "beta_2": 0.999,
24
+ "epsilon": 0.00001
25
+ },
26
+ "max_seq_length": 256,
27
+ "batch_size": 32,
28
+ "num_epochs": 5
29
+ }
sam-artifacts/model.mpk ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1d72b3b82ff9c868352a6f30e31e45e339ab9ae93c6e4ed38684ddd385bddd9
3
+ size 21302041
trainer/.gitignore DELETED
@@ -1 +0,0 @@
1
- /target