diff --git a/ppcls/data/dataloader/pk_sampler.py b/ppcls/data/dataloader/pk_sampler.py
index 04c342a0d..7c3c0d72d 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