megengine.reset_max_memory_stats

reset_max_memory_stats(device=None)

重置内存使用信息

为了保证结果正确， 请先调用 megengine._full_sync before， 然后再调用此接口

上一页

megengine.get_max_allocated_memory

下一页

megengine.set_prealloc_config