Logs a dict containing memory stats to the logger. stats should contain the fields
peak_memory_active, peak_memory_alloc, and peak_memory_reserved as
returned by torchtune.training.get_memory_stats().
Parameters:
stats (Dict[str, float]) – A dictionary containing the peak memory active, peak memory
allocated, and peak memory reserved stats.
To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.