(self, memory, input_size, mem_ratio)
| 77 | return f'LinearMemoryController(target_ratio={self.target_ratio}, available_memory={self.available_memory})' |
| 78 | |
| 79 | def _add_sample(self, memory, input_size, mem_ratio): |
| 80 | self._memory[self._buffer_ptr] = memory |
| 81 | self._input_size[self._buffer_ptr] = input_size |
| 82 | self._mem_ratio[self._buffer_ptr] = mem_ratio |
| 83 | self._buffer_ptr = (self._buffer_ptr + 1) % self.buffer_size |
| 84 | self._buffer_length = min(self._buffer_length + 1, self.buffer_size) |
| 85 | |
| 86 | @contextmanager |
| 87 | def record(self): |