This repository was archived by the owner on May 24, 2021. It is now read-only.
Tags: awwong1/torchprof
Tags
v1.3.0 - Added in support for `profile_memory` (torch >= 1.6), with runtime warning if unsupported - Added in support for `self_cuda_total` (torch >= 1.7), graceful fallback if unsupported - refactored display and profile code into separate files - dynamically populating table columns based on options - table column headings consistent with official names