From ef8f909202f1ce54259b89bbfc54bcbb4855f8ce Mon Sep 17 00:00:00 2001
From: gkaln <55789275+gkaln@users.noreply.github.com>
Date: Wed, 12 May 2021 10:11:31 +0300
Subject: [PATCH] Update train_lstm.py

---
 train_lstm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train_lstm.py b/train_lstm.py
index 2699bcc..c58db2a 100644
--- a/train_lstm.py
+++ b/train_lstm.py
@@ -56,7 +56,7 @@ def test(model, loader, scores, device):
 parser.add_argument('--num_epochs', type=int, default=10, help='number of epochs to train')
 parser.add_argument('--batch_size', type=int, default=16, help='batch size')
 parser.add_argument('--num_workers', type=int, default=0, help='number of workers for data loader; set always to zero!')
-parser.add_argument('--eval_interval', type=int, default=1, help='interval for evaluating models (epochs)')
+parser.add_argument('--eval_interval', type=int, default=50, help='interval for evaluating models (epochs)')
 parser.add_argument('-v', '--verbose', action='store_true', help='show details')
 args = parser.parse_args()
 
@@ -125,4 +125,4 @@ def main():
                 torch.save(pred, os.path.join(args.feats_folder, 'pred-test.pt'))
 
 if __name__ == '__main__':
-    main()
\ No newline at end of file
+    main()