diff --git a/dygraph/slowfast/eval.py b/dygraph/slowfast/eval.py
index 985a8eca07501c08d95ab674ed72cfa84fe1af20..1c229d6bcb14f2dbad13b3d092bbf6bc123f8181 100644
--- a/dygraph/slowfast/eval.py
+++ b/dygraph/slowfast/eval.py
@@ -20,8 +20,8 @@ import logging
 import numpy as np
 import paddle.fluid as fluid
 from paddle.fluid.dygraph.base import to_variable
-from paddle.io import DataLoader, Dataset
-from paddle.incubate.hapi.distributed import DistributedBatchSampler, _all_gather
+from paddle.io import DataLoader, Dataset, DistributedBatchSampler
+from paddle.hapi.model import _all_gather
 from paddle.fluid.dygraph.parallel import ParallelEnv
 
 from model import *
diff --git a/dygraph/slowfast/predict.py b/dygraph/slowfast/predict.py
index 04ad7f0cee781c944beab13d172b66350e587ae7..1a1c8b84d53de1b133bbf1d2f89ef0450422c73b 100644
--- a/dygraph/slowfast/predict.py
+++ b/dygraph/slowfast/predict.py
@@ -21,8 +21,8 @@ import numpy as np
 import json
 import paddle.fluid as fluid
 from paddle.fluid.dygraph.base import to_variable
-from paddle.io import DataLoader, Dataset
-from paddle.incubate.hapi.distributed import DistributedBatchSampler, _all_gather
+from paddle.io import DataLoader, Dataset, DistributedBatchSampler
+from paddle.hapi.model import _all_gather
 from paddle.fluid.dygraph.parallel import ParallelEnv
 
 from model import *
diff --git a/dygraph/slowfast/train.py b/dygraph/slowfast/train.py
index 0a8176590f5e82cea73a19ba09a32e4b93c7c869..48dabcf4b4b1d4daad963785f2fc41373e54ae29 100644
--- a/dygraph/slowfast/train.py
+++ b/dygraph/slowfast/train.py
@@ -24,8 +24,7 @@ import random
 import paddle
 import paddle.fluid as fluid
 from paddle.fluid.dygraph.base import to_variable
-from paddle.io import DataLoader, Dataset
-from paddle.incubate.hapi.distributed import DistributedBatchSampler
+from paddle.io import DataLoader, Dataset, DistributedBatchSampler
 
 from model import *
 from config_utils import *