File tree Expand file tree Collapse file tree 3 files changed +5
-6
lines changed Expand file tree Collapse file tree 3 files changed +5
-6
lines changed Original file line number Diff line number Diff line change 20
20
import numpy as np
21
21
import paddle .fluid as fluid
22
22
from paddle .fluid .dygraph .base import to_variable
23
- from paddle .io import DataLoader , Dataset
24
- from paddle .incubate . hapi .distributed import DistributedBatchSampler , _all_gather
23
+ from paddle .io import DataLoader , Dataset , DistributedBatchSampler
24
+ from paddle .hapi .model import _all_gather
25
25
from paddle .fluid .dygraph .parallel import ParallelEnv
26
26
27
27
from model import *
Original file line number Diff line number Diff line change 21
21
import json
22
22
import paddle .fluid as fluid
23
23
from paddle .fluid .dygraph .base import to_variable
24
- from paddle .io import DataLoader , Dataset
25
- from paddle .incubate . hapi .distributed import DistributedBatchSampler , _all_gather
24
+ from paddle .io import DataLoader , Dataset , DistributedBatchSampler
25
+ from paddle .hapi .model import _all_gather
26
26
from paddle .fluid .dygraph .parallel import ParallelEnv
27
27
28
28
from model import *
Original file line number Diff line number Diff line change 24
24
import paddle
25
25
import paddle .fluid as fluid
26
26
from paddle .fluid .dygraph .base import to_variable
27
- from paddle .io import DataLoader , Dataset
28
- from paddle .incubate .hapi .distributed import DistributedBatchSampler
27
+ from paddle .io import DataLoader , Dataset , DistributedBatchSampler
29
28
30
29
from model import *
31
30
from config_utils import *
You can’t perform that action at this time.
0 commit comments