naghamghanim commited on
Commit
0db1185
โ€ข
1 Parent(s): 7472e65

Upload 2 files

Browse files
Files changed (2) hide show
  1. args.json +73 -0
  2. tag_vocab.pkl +3 -0
args.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_path": "/var/home/nhamad/WojoodData/output-nested",
3
+ "train_path": "/var/home/nhamad/WojoodData/nested/train.txt",
4
+ "val_path": "/var/home/nhamad/WojoodData/nested/val.txt",
5
+ "test_path": "/var/home/nhamad/WojoodData/nested/test.txt",
6
+ "bert_model": "aubmindlab/bert-base-arabertv2",
7
+ "gpus": [
8
+ 0
9
+ ],
10
+ "log_interval": 10,
11
+ "batch_size": 8,
12
+ "num_workers": 0,
13
+ "data_config": {
14
+ "fn": "arabiner.data.datasets.NestedTagsDataset",
15
+ "kwargs": {
16
+ "max_seq_len": 512
17
+ }
18
+ },
19
+ "trainer_config": {
20
+ "fn": "arabiner.trainers.BertNestedTrainer",
21
+ "kwargs": {
22
+ "max_epochs": 50
23
+ }
24
+ },
25
+ "network_config": {
26
+ "fn": "arabiner.nn.BertNestedTagger",
27
+ "kwargs": {
28
+ "dropout": 0.1,
29
+ "bert_model": "aubmindlab/bert-base-arabertv2",
30
+ "num_labels": [
31
+ 4,
32
+ 4,
33
+ 4,
34
+ 4,
35
+ 4,
36
+ 4,
37
+ 4,
38
+ 4,
39
+ 4,
40
+ 4,
41
+ 4,
42
+ 4,
43
+ 4,
44
+ 4,
45
+ 4,
46
+ 4,
47
+ 4,
48
+ 4,
49
+ 4,
50
+ 4,
51
+ 4
52
+ ]
53
+ }
54
+ },
55
+ "optimizer": {
56
+ "fn": "torch.optim.AdamW",
57
+ "kwargs": {
58
+ "lr": 0.0001
59
+ }
60
+ },
61
+ "lr_scheduler": {
62
+ "fn": "torch.optim.lr_scheduler.ExponentialLR",
63
+ "kwargs": {
64
+ "gamma": 1
65
+ }
66
+ },
67
+ "loss": {
68
+ "fn": "torch.nn.CrossEntropyLoss",
69
+ "kwargs": {}
70
+ },
71
+ "overwrite": false,
72
+ "seed": 1
73
+ }
tag_vocab.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0105a59b977eedbc03b7f5aa91ed0474b26af5d8bc96e99632a682cf5aef1896
3
+ size 3796