Skip to content

Commit 22cf383

Browse files
authored
fix slowfast interface bug caused by the movement of hapi dir (#4834)
1 parent 08f3c0b commit 22cf383

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

dygraph/slowfast/eval.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
import numpy as np
2121
import paddle.fluid as fluid
2222
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
2525
from paddle.fluid.dygraph.parallel import ParallelEnv
2626

2727
from model import *

dygraph/slowfast/predict.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@
2121
import json
2222
import paddle.fluid as fluid
2323
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
2626
from paddle.fluid.dygraph.parallel import ParallelEnv
2727

2828
from model import *

dygraph/slowfast/train.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,7 @@
2424
import paddle
2525
import paddle.fluid as fluid
2626
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
2928

3029
from model import *
3130
from config_utils import *

0 commit comments

Comments
 (0)