Bläddra i källkod

Merge branch 'develop' of https://github.com/PaddlePaddle/PaddleX into develop_jf

will-jl944 4 år sedan
förälder
incheckning
cd1627f19a
1 ändrade filer med 1 tillägg och 4 borttagningar
  1. 1 4
      paddlex/cv/models/base.py

+ 1 - 4
paddlex/cv/models/base.py

@@ -20,7 +20,6 @@ import copy
 import math
 import yaml
 import json
-import numpy as np
 import paddle
 from paddle.io import DataLoader, DistributedBatchSampler
 from paddleslim import QAT
@@ -251,9 +250,7 @@ class BaseModel:
             collate_fn=dataset.batch_transforms,
             num_workers=dataset.num_workers,
             return_list=True,
-            use_shared_memory=use_shared_memory,
-            worker_init_fn=lambda worker_id: np.random.seed(np.random.get_state()[1][0] + worker_id)
-        )
+            use_shared_memory=use_shared_memory)
 
         return loader