Skip to content

Commit 5d2030a

Browse files
authored
[FIX] adapt to paddle 2.4 & 2.5 (#3316)
1 parent 61afee2 commit 5d2030a

File tree

2 files changed

+17
-2
lines changed

2 files changed

+17
-2
lines changed

contrib/CrossPseudoSupervision/batch_transforms/custom_collate.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,14 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from paddle.fluid.dataloader.collate import default_collate_fn
15+
import paddle
16+
paddle_version = paddle.__version__[:3]
17+
# paddle version < 2.5.0 and not develop
18+
if paddle_version not in ["2.5", "0.0"]:
19+
from paddle.fluid.dataloader.collate import default_collate_fn
20+
# paddle version >= 2.5.0 or develop
21+
else:
22+
from paddle.io.dataloader.collate import default_collate_fn
1623

1724

1825
class SegCollate(object):

contrib/PanopticSeg/paddlepanseg/datasets/base_dataset.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,19 @@
1818

1919
import numpy as np
2020
import paddle
21-
from paddle.fluid.dataloader.collate import default_collate_fn
21+
2222
from paddleseg.transforms import Compose
2323

2424
from paddlepanseg.cvlibs import build_info_dict
2525

26+
paddle_version = paddle.__version__[:3]
27+
# paddle version < 2.5.0 and not develop
28+
if paddle_version not in ["2.5", "0.0"]:
29+
from paddle.fluid.dataloader.collate import default_collate_fn
30+
# paddle version >= 2.5.0 or develop
31+
else:
32+
from paddle.io.dataloader.collate import default_collate_fn
33+
2634

2735
class PanopticDataset(paddle.io.Dataset):
2836
NO_COLLATION_KEYS = ('gt_fields', 'trans_info', 'image_id')

0 commit comments

Comments
 (0)