(self, batch)
| 27 | self.batch = None |
| 28 | |
| 29 | def set_batch(self, batch): |
| 30 | if self.batch is not None: |
| 31 | if isinstance(self.batch, torch.Tensor): |
| 32 | assert self.batch.shape[1:] == batch.shape[1:], "Check: shapes probably should not change during training" |
| 33 | |
| 34 | self.batch = batch |
| 35 | |
| 36 | def get_batch(self, x=None): |
| 37 | assert self.batch is not None, "Error: need to set a batch first" |
nothing calls this directly
no outgoing calls
no test coverage detected