Update text

pull/10/head
Eren G 2018-07-30 14:09:50 +02:00
parent 3ad5aea93b
commit bce43f4d6e
1 changed files with 1 additions and 1 deletions

View File

@ -60,7 +60,7 @@ class TacotronTrainTest(unittest.TestCase):
count = 0
for param, param_ref in zip(model.parameters(), model_ref.parameters()):
# ignore pre-higway layer since it works conditional
if count not in [145, 59]:
if count not in [148, 59]:
assert (param != param_ref).any(), "param {} with shape {} not updated!! \n{}\n{}".format(count, param.shape, param, param_ref)
count += 1