(self, room: int)
| 315 | return save_path |
| 316 | |
| 317 | def release_memory(self, room: int): |
| 318 | if room in self._rdma_buffers: |
| 319 | self._rdma_buffers.pop(room, None) |
| 320 | torch.cuda.empty_cache() |
| 321 | |
| 322 | def remove(self, room: int): |
| 323 | self.release_memory(room) |