From a96a3efaad023c3f4f95ea856a45145f9f7675a1 Mon Sep 17 00:00:00 2001 From: mohammad Date: Fri, 20 Jan 2023 14:11:34 -0500 Subject: [PATCH] enable sink mode explicitly due to change in default value --- tests/st/dataset/test_gpu_reset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/st/dataset/test_gpu_reset.py b/tests/st/dataset/test_gpu_reset.py index 9226874f9a5..cd93204f66f 100644 --- a/tests/st/dataset/test_gpu_reset.py +++ b/tests/st/dataset/test_gpu_reset.py @@ -91,7 +91,7 @@ def test_dataset_reset_sink(fast_recovery, num_parallel_workers, python_multipro num_epochs = 3 reset_point = 2 # 2nd epoch 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)