src/transformers/trainer_utils.py/TrainerMemoryTracker/cpu_mem_used
class TrainerMemoryTracker:    def cpu_mem_used(self):
        """get resident set size memory for the current process"""
        return self.process.memory_info().rss