(self)
| 91 | return len(self.data) |
| 92 | |
| 93 | def load_data(self): |
| 94 | max_value = int(2 ** self.msg_len) |
| 95 | if not hasattr(self, 'b2t_maps'): |
| 96 | rand_tokens = torch.randperm(CLIP_TOKEN_MAX) + 1 |
| 97 | self.b2t_maps = rand_tokens[:2 * self.msg_len].reshape(self.msg_len, 2) |
| 98 | |
| 99 | if not hasattr(self, 'data'): |
| 100 | dec_messages = sample_K_from_N(self.max_size, max_value) |
| 101 | self.data = [self.dec_message_to_tokens(x) for x in dec_messages] |
| 102 | |
| 103 | # Decimal message -> Binary message + CLIP tokens |
| 104 | def dec_message_to_tokens(self, dec_message): |
no test coverage detected