Skip to content

Commit

Permalink
update for training
Browse files Browse the repository at this point in the history
  • Loading branch information
lukhy committed Jun 20, 2019
1 parent b5209eb commit 6ab2a79
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/conv.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ def __init__(self, vocabulary, blank=0, name="masr"):
modules.append(ConvBlock(nn.Conv1d(161, 500, 48, 2, 97), 0.2))

for i in range(7):
modules.append(ConvBlock(nn.Conv1d(250, 500, 7, 1), 0.4))
modules.append(ConvBlock(nn.Conv1d(250, 500, 7, 1), 0.3))

modules.append(ConvBlock(nn.Conv1d(250, 2000, 32, 1), 0.5))

Expand Down

0 comments on commit 6ab2a79

Please sign in to comment.