Skip to content

Commit 8ba3290

Browse files
committed
no message
1 parent f402d46 commit 8ba3290

11 files changed

+35
-26
lines changed

.idea/workspace.xml

+18-12
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

__pycache__/conf.cpython-36.pyc

27 Bytes
Binary file not shown.

__pycache__/data.cpython-36.pyc

0 Bytes
Binary file not shown.

conf.py

+6-5
Original file line numberDiff line numberDiff line change
@@ -14,15 +14,16 @@
1414
d_model = 512
1515
n_layers = 6
1616
n_heads = 8
17-
ffn_hidden = 256
17+
ffn_hidden = 2048
1818
drop_prob = 0.1
1919

2020
# optimizer parameter setting
21-
init_lr = 1e-4
22-
factor = 0.8
21+
init_lr = 1e-5
22+
factor = 0.9
23+
min_lr = init_lr * 1e-12
2324
patience = 10
24-
warmup = 5
25+
warmup = 100
2526
weight_decay = 5e-4
26-
epoch = 2000
27+
epoch = 1000
2728
clip = 1
2829
inf = float('inf')

data.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from util.tokenizer import Tokenizer
99

1010
tokenizer = Tokenizer()
11-
loader = DataLoader(ext=('.de', '.en'),
11+
loader = DataLoader(ext=('.en', '.de'),
1212
tokenize_en=tokenizer.tokenize_en,
1313
tokenize_de=tokenizer.tokenize_de,
1414
init_token='<sos>',

graph.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def read(name):
1515
file = re.sub('\\]', '', file)
1616
f.close()
1717

18-
return [float(i) for idx, i in enumerate(file.split(',')) if idx <= 150]
18+
return [float(i) for idx, i in enumerate(file.split(','))]
1919

2020

2121
def draw():

image/train_result.jpg

-40.6 KB
Loading

0 commit comments

Comments
 (0)