Skip to content

Commit 37729f0

Browse files
fixing test (#451)
1 parent 446a1b5 commit 37729f0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/test_cpu_models.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -299,6 +299,8 @@ def test_tbptt_cpu_model():
299299
"""
300300
testing_utils.reset_seed()
301301

302+
save_dir = testing_utils.init_save_dir()
303+
302304
truncated_bptt_steps = 2
303305
sequence_size = 30
304306
batch_size = 30
@@ -366,6 +368,8 @@ def train_dataloader(self):
366368

367369
assert result == 1, 'training failed to complete'
368370

371+
testing_utils.clear_save_dir()
372+
369373

370374
def test_single_gpu_model():
371375
"""

0 commit comments

Comments
 (0)