Update config
Browse files
config.py
CHANGED
|
@@ -5,7 +5,7 @@ bert_lr = 3e-06
|
|
| 5 |
cased = False
|
| 6 |
parser_lr = 1e-06
|
| 7 |
tagger_lr = 8e-05
|
| 8 |
-
device = 'cuda:0'
|
| 9 |
early_stop = 5
|
| 10 |
epochs = 50
|
| 11 |
expand_iota = True
|
|
@@ -24,7 +24,7 @@ print_gold = True
|
|
| 24 |
scheduler = 'linear'
|
| 25 |
seed = 3
|
| 26 |
subword_prefix = '##'
|
| 27 |
-
test_path = '
|
| 28 |
-
train_path = '
|
| 29 |
-
val_path = '
|
| 30 |
vocabs_path = 'vocabs/ud_proiel.json'
|
|
|
|
| 5 |
cased = False
|
| 6 |
parser_lr = 1e-06
|
| 7 |
tagger_lr = 8e-05
|
| 8 |
+
device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
|
| 9 |
early_stop = 5
|
| 10 |
epochs = 50
|
| 11 |
expand_iota = True
|
|
|
|
| 24 |
scheduler = 'linear'
|
| 25 |
seed = 3
|
| 26 |
subword_prefix = '##'
|
| 27 |
+
test_path = 'data/ud_proiel/grc_proiel-ud-test.conllu'
|
| 28 |
+
train_path = 'data/ud_proiel/grc_proiel-ud-train.conllu'
|
| 29 |
+
val_path = 'data/ud_proiel/grc_proiel-ud-dev.conllu'
|
| 30 |
vocabs_path = 'vocabs/ud_proiel.json'
|