Code

Pytorch integration: Remat#10

Code

Open

arnavsinghvi11 wants to merge 5 commits intoShishirPatil:mainShishirPatil/poet:mainfrom arnavsinghvi11:remat_and_pagingarnavsinghvi11/poet:remat_and_pagingCopy head branch name to clipboard

+496-3

Commits

Commits on Feb 22, 2023

• network transformation function & testing for PyTorch to POET

  

  Arnav Singhvi

  committed
  853e088

  Copy full SHA for 853e088

• network transformation function & testing for TensorFlow to POET

  

  Arnav Singhvi

  committed
  06349a8

  Copy full SHA for 06349a8

• graph transformation function for converting PyTorch model graph to POET layers graph

  

  Arnav Singhvi

  committed
  305f861

  Copy full SHA for 305f861

Commits on Feb 23, 2023

• implemented rematerialization and paging for PyTorch models, currently only remat completes training

  

  Arnav Singhvi

  committed
  b5531a0

  Copy full SHA for b5531a0

Commits on Apr 4, 2023

• added comprehensive comments for remat functions

  arnavsinghvi11

  committed
  906aab6

  Copy full SHA for 906aab6