(self)
| 143 | return self.num_samples |
| 144 | |
| 145 | def state_dict(self) -> dict[str, int]: |
| 146 | return { |
| 147 | 'epoch': self.epoch, |
| 148 | 'idx': self.idx, |
| 149 | } |
| 150 | |
| 151 | def load_state_dict(self, state_dict): |
| 152 | self.epoch = state_dict['epoch'] |
no outgoing calls
no test coverage detected