kdudzic/en-ner-conll-2003
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'train.py'

Browse Source
This commit is contained in:
Kacper Dudzic 2022-06-07 21:36:55 +02:00
parent 43847f486b
commit 1dd798e29b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -103,7 +103,7 @@ optimizer = torch.optim.Adam(ner_model.parameters())
# TRAIN # TRAIN
print('-----TRAINING-----') print('-----TRAINING-----')
for epoch in range(1): for epoch in range(5):
loss_score = 0 loss_score = 0
acc_score = 0 acc_score = 0
prec_score = 0 prec_score = 0