(self)
| 530 | return self.total_size - (self.last_iter+1)*self.batch_size |
| 531 | |
| 532 | def save(self): |
| 533 | torch.save({'task_name': self.task_name, |
| 534 | 'task_size': self.world_size, |
| 535 | 'task_rank': self.rank, |
| 536 | 'ret_file': self.indices}, self.this_ret_path) |
| 537 | printlog("save sampler file ------> {}".format(self.this_ret_path)) |
| 538 | |
| 539 | |
| 540 | class RandomIdentitySampler(Sampler): |