diff --git a/train.py b/train.py
index b269889..c5c1b37 100644
--- a/train.py
+++ b/train.py
@@ -103,7 +103,7 @@ optimizer = torch.optim.Adam(ner_model.parameters())
 
 # TRAIN
 print('-----TRAINING-----')
-for epoch in range(1):
+for epoch in range(5):
     loss_score = 0
     acc_score = 0
     prec_score = 0