From 46763e76c20a4e0973e1ea7553b0828782340d24 Mon Sep 17 00:00:00 2001 From: SzamanFL Date: Sun, 24 Jan 2021 17:16:15 +0100 Subject: [PATCH] Small change --- src/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/train.py b/src/train.py index ca6abe4..b04ade8 100644 --- a/src/train.py +++ b/src/train.py @@ -176,6 +176,6 @@ def main(): if args.decoder: decoder.load_state_dict(torch.load(args.decoder)) - train_iterate(pairs, encoder, decoder, 50000, input_vocab, target_vocab) + train_iterate(pairs, encoder, decoder, 70000, input_vocab, target_vocab) main()