diff --git a/configs/faster_reader.yml b/configs/faster_reader.yml index 643b4721ab12dac2f8806471b562df3b52abb2f4..26835d4c6d2ef6187cd1b5ce23283cf049194ba8 100644 --- a/configs/faster_reader.yml +++ b/configs/faster_reader.yml @@ -24,6 +24,10 @@ TrainReader: - !Permute to_bgr: false channel_first: true + batch_transforms: + - !PadBatch + pad_to_stride: -1. + use_padded_im_info: false batch_size: 1 shuffle: true worker_num: 2 diff --git a/configs/mask_reader.yml b/configs/mask_reader.yml index 50432ece63e053989ef05b9e2656e4c171723eae..31742f2057f4a86d6a4a6d5f6d5bc6ca7dfa4b54 100644 --- a/configs/mask_reader.yml +++ b/configs/mask_reader.yml @@ -25,6 +25,10 @@ TrainReader: - !Permute to_bgr: false channel_first: true + batch_transforms: + - !PadBatch + pad_to_stride: -1. + use_padded_im_info: false batch_size: 1 shuffle: true worker_num: 2