s426135/wmt-2020-pl-en
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Small change

Browse Source
This commit is contained in:
SzamanFL 2021-01-24 17:16:15 +01:00
parent f0ff04b6ac
commit 46763e76c2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/train.py
View File

@ -176,6 +176,6 @@ def main():
if args.decoder: if args.decoder:
decoder.load_state_dict(torch.load(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() main()