cfb7702
1
2
3
4
5
6
7
8
9
10
11
import torch from torch.utils.data import Dataset default_sub_data_config = {} class Joint3D(Dataset): def __init__(self, sub_data_config: dict) -> None: super().__init__() self.sub_data_config = sub_data_config
