diff --git a/ppcls/data/dataloader/pk_sampler.py b/ppcls/data/dataloader/pk_sampler.py
index 04c342a0db19b60624957976d6a37c5ee118b8a8..7c3c0d72df8f63f7f0d21eea82ecd5da18bd7914 100644
--- a/ppcls/data/dataloader/pk_sampler.py
+++ b/ppcls/data/dataloader/pk_sampler.py
@@ -18,9 +18,7 @@ from __future__ import division
 from collections import defaultdict
 
 import numpy as np
-import paddle.distributed as dist
 from paddle.io import DistributedBatchSampler
-
 from ppcls.utils import logger