Code

Introduce CheckpointIO Plugin#8743

Code

Merged

SeanNaren merged 40 commits intomasterLightning-AI/pytorch-lightning:masterfrom feat/ckpt_pluginLightning-AI/pytorch-lightning:feat/ckpt_pluginCopy head branch name to clipboardAug 13, 2021

+420-46

Commits

Commits on Jul 19, 2021

• poc API

  

  SeanNaren

  committed
  a93e452

  Copy full SHA for a93e452

Commits on Aug 5, 2021

• Merge branch 'master' into feat/ckpt_plugin

  

  SeanNaren

  committed
  72f4dfd

  Copy full SHA for 72f4dfd

• Fix up the API, unsure on connection

  

  SeanNaren

  committed
  e7d2b66

  Copy full SHA for e7d2b66

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for b41e794
  pre-commit-ci[bot]

  committed
  b41e794

  Copy full SHA for b41e794

• Example API

  

  SeanNaren

  committed
  9161980

  Copy full SHA for 9161980

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 7aa4e8c
  pre-commit-ci[bot]

  committed
  7aa4e8c

  Copy full SHA for 7aa4e8c

• Update all constructors

  

  SeanNaren

  committed
  dffe088

  Copy full SHA for dffe088

Commits on Aug 6, 2021

• Move towards having the checkpoint plugin not require the plugin, and focus on IO

  

  SeanNaren

  committed
  cacf0e5

  Copy full SHA for cacf0e5

• Remove import

  

  SeanNaren

  committed
  028ac38

  Copy full SHA for 028ac38

• Fix tests

  

  SeanNaren

  committed
  99c7a46

  Copy full SHA for 99c7a46

• Change name

  

  SeanNaren

  committed
  3adc486

  Copy full SHA for 3adc486

Commits on Aug 9, 2021

• Cleanups

  

  SeanNaren

  committed
  b7d5b55

  Copy full SHA for b7d5b55

• Fixes/Cleanups

  

  SeanNaren

  committed
  0a0a068

  Copy full SHA for 0a0a068

• Use property

  

  SeanNaren

  committed
  97fb2a2

  Copy full SHA for 97fb2a2

• Fixes to signature

  

  SeanNaren

  committed
  402156e

  Copy full SHA for 402156e

• Merge branch 'master' into feat/ckpt_plugin

  

  SeanNaren

  committed
  5310a7f

  Copy full SHA for 5310a7f

• Add warning for TPU plugins that they do not support custom checkpoint plugins, add rudimentary test

  

  SeanNaren

  committed
  d7f567a

  Copy full SHA for d7f567a

Commits on Aug 10, 2021

• Cleanup API, introduce storage options

  

  SeanNaren

  committed
  fcc24b4

  Copy full SHA for fcc24b4

• Update signature to be more general

  

  SeanNaren

  committed
  4421276

  Copy full SHA for 4421276

Commits on Aug 11, 2021

• Address feedback, add test for support check

  

  SeanNaren

  committed
  d84cce1

  Copy full SHA for d84cce1

• Merge branch 'master' into feat/ckpt_plugin

  

  SeanNaren

  committed
  38c22a2

  Copy full SHA for 38c22a2

• Add CHANGELOG.md

  

  SeanNaren

  committed
  b7f37ee

  Copy full SHA for b7f37ee

• fix tests

  

  SeanNaren

  committed
  49086cc

  Copy full SHA for 49086cc

• change name

  

  SeanNaren

  committed
  936f65a

  Copy full SHA for 936f65a

• Fix mypy

  

  SeanNaren

  committed
  049a676

  Copy full SHA for 049a676

Commits on Aug 12, 2021

• Reviews

  

  SeanNaren

  committed
  1ff0912

  Copy full SHA for 1ff0912

• Add ability to pass checkpoint plugin through the trainer

  

  SeanNaren

  committed
  1841d3b

  Copy full SHA for 1841d3b

• Add constraints

  

  SeanNaren

  committed
  b909dfe

  Copy full SHA for b909dfe

• Match signature to see if mypy works

  

  SeanNaren

  committed
  50b11b5

  Copy full SHA for 50b11b5

Commits on Aug 13, 2021

• Address review points

  

  SeanNaren

  committed
  9e16e34

  Copy full SHA for 9e16e34

• Revert changes to typing

  

  SeanNaren

  committed
  642e6fa

  Copy full SHA for 642e6fa

• Add docs/doc strings and API

  

  SeanNaren

  committed
  5c9e973

  Copy full SHA for 5c9e973

• Address feedback

  

  SeanNaren

  committed
  6361c87

  Copy full SHA for 6361c87

• Update pytorch_lightning/plugins/training_type/training_type_plugin.py

  Show description for c921dbb
  

  Sean Naren

  and

  carmocca

  authored
  c921dbb

  Copy full SHA for c921dbb

• Address reviews

  

  SeanNaren

  committed
  fd82276

  Copy full SHA for fd82276

• Update typing

  

  SeanNaren

  committed
  2fc3558

  Copy full SHA for 2fc3558

• Refactor name

  

  SeanNaren

  committed
  21783f6

  Copy full SHA for 21783f6

• Clear up signature of function; checkpoint_plugin -> checkpoint_io

  

  SeanNaren

  committed
  3b8c3f5

  Copy full SHA for 3b8c3f5

• Slightly cleaner

  

  SeanNaren

  committed
  9cfe98f

  Copy full SHA for 9cfe98f

• Address reviews

  

  SeanNaren

  committed
  8f234e0

  Copy full SHA for 8f234e0