Skip to content

Commit 988c426

Browse files
committed
Minor bugfixes
- A typo is fixed in config.json - "self.self.scorer" typo is fixed in single_model_trainer.py - Added an environmental variable thing for debugging cuda errors to main.py
1 parent 5f6ec17 commit 988c426

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

config/config.json

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
"dataset_properties": {
33
"data_path": "./data/turkish_test.DUMP",
44
"batch_size": 32,
5-
"stop_word_path": "D:/Anaconda3/nltk_data/corpora/stopwords/turkish",
5+
"stop_word_path": "./data/stopwords/turkish",
66
"embedding_vector": "fasttext.tr.300d",
77
"vector_cache": "./data/fasttext",
88
"pretrained_embedding_path": "./data/fasttext/wiki.tr",
@@ -120,12 +120,12 @@
120120
"training_properties": {
121121
"learner": "lstmcrf",
122122
"task": "ner",
123-
"optimizer": "Padam",
123+
"optimizer": "SGD",
124124
"learning_rate": 0.05,
125125
"final_learning_rate": 0.1,
126126
"scheduler_type": "cos",
127127
"amsgrad": true,
128-
"amsbound0": false,
128+
"amsbound": false,
129129
"partial_adam": 0.25,
130130
"weight_decay": 0,
131131
"momentum": 0.9,

main.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@
1919
from training.trainer import Trainer
2020
from utils.utils import save_vocabulary
2121

22+
os.environ['CUDA_LAUNCH_BLOCKING'] = "1"
23+
2224
logging.config.fileConfig(fname='./config/config.logger', disable_existing_loggers=False)
2325
logger = logging.getLogger("Main")
2426

training/single_model_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def train_iters(self, model, checkpoint=None):
119119
input=self.dev_iter,
120120
criterion=criterion,
121121
save_path=self.save_path,
122-
scorer=self.self.scorer)
122+
scorer=self.scorer)
123123
if best_vali_acc < vali_accuracy:
124124
best_vali_loss = vali_loss
125125
best_vali_acc = vali_accuracy

0 commit comments

Comments
 (0)