Selaa lähdekoodia

Merge pull request #1065 from will-jl944/bug_fix

use default worker_init_fn for dataloader
will-jl944 4 vuotta sitten
vanhempi
commit
fd40046d7f
1 muutettua tiedostoa jossa 1 lisäystä ja 4 poistoa
  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