Code

Use xm.save to save model on TPU#3044

Code

Closed

lezwon wants to merge 44 commits intoLightning-AI:masterLightning-AI/pytorch-lightning:masterfrom lezwon:bugfix/2700_xm_savelezwon/pytorch-lightning:bugfix/2700_xm_saveCopy head branch name to clipboard

+43-27

Commits

Commits on Sep 19, 2020

• support xm save

  lezwon

  committed
  7bdc8ae

• or to and

  lezwon

  committed
  131ec3c

• removed rank_zero_only from on_train_start

  lezwon

  committed
  df49e5c

• delete file if exists

  Show description for 43d55b6
  lezwon

  committed
  43d55b6

• added rank_zero_only for _del_model

  lezwon

  committed
  1313940

• load state_dict

  lezwon

  committed
  6d2eb42

• access state_dict from checkpoint

  lezwon

  committed
  1ee1b05

• bytesbuffer to filepath

  lezwon

  committed
  f62c9e5

• added checkpoint_callback check

  lezwon

  committed
  5c696b9

• print function name

  lezwon

  committed
  82f32b7

• return after saving xla tensor

  lezwon

  committed
  3449f93

• log global rank

  lezwon

  committed
  9568af3

• removed trainer from check_monitor_top_k

  lezwon

  committed
  52e4b7c

• print current device

  lezwon

  committed
  702f0b9

• rank_zero_only for _do_check_save

  lezwon

  committed
  8746488

• log in atomic save

  lezwon

  committed
  a123088

• log in _do_check_save

  lezwon

  committed
  7a0b6e5

• log in check_monitor_top_k

  lezwon

  committed
  13dcfa1

• more logs inside check_monitor_top_k

  lezwon

  committed
  189cc3b

• remove rank zero only form _do_check_save

  lezwon

  committed
  ca1da3a

• added is_global_zero in _do_check_save

  lezwon

  committed
  aaecd95

• fix del_list issue

  lezwon

  committed
  f302d24

• added global condition check

  lezwon

  committed
  04ce20e

• added pdb

  lezwon

  committed
  2032128

• removed pdb

  lezwon

  committed
  0d1611e

• log best_k_models

  lezwon

  committed
  683a1b7

• more changes

  lezwon

  committed
  d81e61a

• log after save

  lezwon

  committed
  99ee490

• fix error

  lezwon

  committed
  dfdcd76

• fix error again

  lezwon

  committed
  ca303bc

• fix error again and again

  lezwon

  committed
  de4a117

• remove device call

  lezwon

  committed
  31e867d

• ayee add this too. done saving part

  lezwon

  committed
  15eb2b8

• remove state dict

  lezwon

  committed
  8784d59

• remove inspect statements

  lezwon

  committed
  79882f0

• print stack trace if not is_xla_tensor

  lezwon

  committed
  b20855b

• remove inspect reference

  lezwon

  committed
  a4f2c6b

• Revert "print stack trace if not is_xla_tensor"

  Show description for 8f41335
  lezwon

  committed
  8f41335

• removed state dict

  lezwon

  committed
  9dbfdbf

• added is_xla_tensor into docstring

  lezwon

  committed
  85c5b55

• save only state dict

  lezwon

  committed
  d488119

• save entire checkpoint

  lezwon

  committed
  12d675d

• load state dict

  lezwon

  committed
  c1d8f74

• xla check inside atomic save

  lezwon

  committed
  c883572