|
@@ -20,7 +20,6 @@ import copy
|
|
|
import math
|
|
import math
|
|
|
import yaml
|
|
import yaml
|
|
|
import json
|
|
import json
|
|
|
-import numpy as np
|
|
|
|
|
import paddle
|
|
import paddle
|
|
|
from paddle.io import DataLoader, DistributedBatchSampler
|
|
from paddle.io import DataLoader, DistributedBatchSampler
|
|
|
from paddleslim import QAT
|
|
from paddleslim import QAT
|
|
@@ -251,9 +250,7 @@ class BaseModel:
|
|
|
collate_fn=dataset.batch_transforms,
|
|
collate_fn=dataset.batch_transforms,
|
|
|
num_workers=dataset.num_workers,
|
|
num_workers=dataset.num_workers,
|
|
|
return_list=True,
|
|
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
|
|
return loader
|
|
|
|
|
|