fixing test (#451)
This commit is contained in:
parent
446a1b5d45
commit
37729f0a17
|
@ -299,6 +299,8 @@ def test_tbptt_cpu_model():
|
|||
"""
|
||||
testing_utils.reset_seed()
|
||||
|
||||
save_dir = testing_utils.init_save_dir()
|
||||
|
||||
truncated_bptt_steps = 2
|
||||
sequence_size = 30
|
||||
batch_size = 30
|
||||
|
@ -366,6 +368,8 @@ def test_tbptt_cpu_model():
|
|||
|
||||
assert result == 1, 'training failed to complete'
|
||||
|
||||
testing_utils.clear_save_dir()
|
||||
|
||||
|
||||
def test_single_gpu_model():
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue