Home
last modified time | relevance | path

Searched refs:save_checkpoint (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcheckpointing_test.py41 save_checkpoint = trackable_utils.Checkpoint(v=variables_lib.Variable(
43 save_path = save_checkpoint.save(
/external/pytorch/docs/source/elastic/
Dtrain_script.rst18 ``save_checkpoint(path)`` logic in your script. When any number of
47 save_checkpoint(state)
/external/tensorflow/tensorflow/python/checkpoint/
Dcheckpoint_test.py777 save_checkpoint = trackable_utils.Checkpoint(v=variables_lib.Variable(1.))
781 save_path = save_checkpoint.write(checkpoint_prefix)
784 self.evaluate([save_checkpoint.v.initializer])
794 self.evaluate(save_checkpoint.v.assign(3.))
796 self.evaluate(save_checkpoint.v.assign(0.))
810 save_checkpoint = trackable_utils.Checkpoint(v=variables_lib.Variable(1.))
814 save_path = save_checkpoint.write(prefix)
817 self.evaluate([save_checkpoint.v.initializer])
827 self.evaluate(save_checkpoint.v.assign(3.))
829 self.evaluate(save_checkpoint.v.assign(0.))
[all …]
/external/pytorch/functorch/examples/dp_cifar10/
Dcifar10_opacus.py36 def save_checkpoint(state, is_best, filename="checkpoint.tar"): function
256 save_checkpoint(
Dcifar10_transforms.py36 def save_checkpoint(state, is_best, filename="checkpoint.tar"): function
292 save_checkpoint(
/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
Dlookup_ops_test.py1989 save_checkpoint = trackable.Checkpoint(table=save_table)
1996 save_path = save_checkpoint.save(save_prefix)
1997 del save_table, save_checkpoint