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()