ソースを参照

use default worker_init_fn for dataloader

will-jl944 4 年 前
コミット
41aa537bda
1 ファイル変更1 行追加4 行削除
  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