Method __init__

datasets/multi_resolution_sampler.py:12–22

(
        self,
        ori_thw_list: List[Tuple],
        dp_size: int,
        rnd_state: np.random.RandomState,
        bucket_config: BucketConfig,
    )

class BucketFactory:

    def __init__(
        self,
        ori_thw_list: List[Tuple],
        dp_size: int,
        rnd_state: np.random.RandomState,
        bucket_config: BucketConfig,
    ) -> None:
        self.ori_thw_list = ori_thw_list
        self.dp_size = dp_size
        self.rnd_state = rnd_state
        self.bucket_config = bucket_config

    def __call__(self, ) -> Any:
        flop_list = [