Published February 20, 2023 | Version v1
Software Open

Mimose_info_for_IPDPS23

Creators

  • 1. Beihang University
  • 2. SenseTime Research

Description

This repository contains the implementation of **Mimose**, our input-aware tensor checkpointing planner described in the paper "Exploiting Input Tensor Dynamics in Activation Checkpointing for Efficient Training on GPU" (to be appeared in IPDPS'23).

Files

Mimose-IPDPS23-Open-Source.zip

Files (18.9 MB)

Name Size Download all
md5:9698f7eea68eb6c2ebb396cb529a9b3f
18.9 MB Preview Download