(self, args)
| 37 | class MeshTokenizer(nn.Module): |
| 38 | |
| 39 | def __init__(self, args): |
| 40 | super().__init__() |
| 41 | self.pad_id = -1 |
| 42 | self.num_discrete_coors = args.n_discrete_size # default: 800 |
| 43 | self.codebook_size = args.n_discrete_size # default: 128 |
| 44 | self.coor_continuous_range = (-1., 1.) |
| 45 | |
| 46 | |
| 47 | def tokenize(self, data_dict: dict) -> dict: |
nothing calls this directly
no outgoing calls
no test coverage detected