p85971406
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48084 Fix: enable sink mode explicitly in dataset reset smoke test 

Merge pull request !48084 from Mohammad Motallebi/fix_reset_smoke_sink
This commit is contained in:
i-robot 2023-01-23 18:41:37 +00:00 committed by Gitee
parent 0dcd3a8a54 a96a3efaad
commit a5ada6bb96
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/dataset/test_gpu_reset.py
View File

@ -91,7 +91,7 @@ def test_dataset_reset_sink(fast_recovery, num_parallel_workers, python_multipro
num_epochs = 3 num_epochs = 3
reset_point = 2 # 2nd epoch reset_point = 2 # 2nd epoch
cb = MyCallback(dataset_size=data.get_dataset_size(), reset_point=reset_point) cb = MyCallback(dataset_size=data.get_dataset_size(), reset_point=reset_point)
model.train(num_epochs, data, callbacks=[cb]) model.train(num_epochs, data, callbacks=[cb], dataset_sink_mode=True)
ds.config.set_fast_recovery(original_fast_recovery) ds.config.set_fast_recovery(original_fast_recovery)