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

Eddited max length

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

2
src/predict.py
View File

@ -12,7 +12,7 @@ from Encoder import Encoder
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
SOS=0 SOS=0
EOS=1 EOS=1
MAX_LENGTH=25 MAX_LENGTH=40
def clear_line(string): def clear_line(string):
string = ''.join( string = ''.join(

2
src/train.py
View File

@ -13,7 +13,7 @@ from Vocab import Vocab
from Encoder import Encoder from Encoder import Encoder
from Decoder import Decoder from Decoder import Decoder
MAX_LENGTH = 25 MAX_LENGTH = 40
SOS=0 SOS=0
EOS=1 EOS=1
teacher_forcing_ratio=0.5 teacher_forcing_ratio=0.5